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
530 B
YAML
20 lines
530 B
YAML
model:
|
|
class_path: anomalib.models.Uflow
|
|
init_args:
|
|
flow_steps: 4
|
|
permute_soft: false
|
|
affine_clamp: 2.0
|
|
affine_subnet_channels_ratio: 1.0
|
|
backbone: mcait # official: mcait, other extractors tested: resnet18, wide_resnet50_2. Could use others...
|
|
|
|
# PL Trainer Args. Don't add extra parameter here.
|
|
trainer:
|
|
max_epochs: 200
|
|
log_every_n_steps: 50
|
|
callbacks:
|
|
- class_path: lightning.pytorch.callbacks.EarlyStopping
|
|
init_args:
|
|
patience: 20
|
|
monitor: pixel_AUROC
|
|
mode: max
|