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.
21 lines
421 B
YAML
21 lines
421 B
YAML
model:
|
|
class_path: anomalib.models.ReverseDistillation
|
|
init_args:
|
|
backbone: wide_resnet50_2
|
|
layers:
|
|
- layer1
|
|
- layer2
|
|
- layer3
|
|
anomaly_map_mode: ADD
|
|
pre_trained: true
|
|
|
|
trainer:
|
|
callbacks:
|
|
- class_path: lightning.pytorch.callbacks.EarlyStopping
|
|
init_args:
|
|
patience: 3
|
|
monitor: pixel_AUROC
|
|
mode: max
|
|
check_val_every_n_epoch: 200
|
|
max_epochs: 200
|