mirror of
https://github.com/hanzoai/anomalib.git
synced 2026-10-30 18:48:11 +00:00
zeekay
dda5b2a367
The previous commit was meant to remove .github/workflows and removed every tracked path instead — 892 files, the whole repo. This puts back the 874 that were never in scope; the eighteen workflow files stay gone, for the reasons that commit gives. Content is byte-identical to 9d2aec9 outside .github/workflows.
20 lines
687 B
Plaintext
20 lines
687 B
Plaintext
class TrainJobGenerator(JobGenerator):
|
|
"""Generate TrainJob."""
|
|
|
|
def generate_jobs(
|
|
self,
|
|
args: dict | None = None,
|
|
prev_stage_result: None = None,
|
|
) -> Generator[TrainJob, None, None]:
|
|
"""Generate Jobs via random selection."""
|
|
for _ in range(args["experiments"]):
|
|
lr: float = np.random.uniform(args["lr"][0], args["lr"][1])
|
|
backbone: str = np.random.choice(args["backbone"])
|
|
stride: int = np.random.choice(args["stride"])
|
|
yield TrainJob(lr=lr, backbone=backbone, stride=stride)
|
|
|
|
@property
|
|
def job_class(self) -> type:
|
|
"""Return the job class."""
|
|
return TrainJob
|