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.
26 lines
508 B
YAML
26 lines
508 B
YAML
model:
|
|
class_path: anomalib.models.Cflow
|
|
init_args:
|
|
backbone: wide_resnet50_2
|
|
layers:
|
|
- layer2
|
|
- layer3
|
|
- layer4
|
|
pre_trained: true
|
|
fiber_batch_size: 64
|
|
decoder: freia-cflow
|
|
condition_vector: 128
|
|
coupling_blocks: 8
|
|
clamp_alpha: 1.9
|
|
permute_soft: false
|
|
lr: 0.0001
|
|
|
|
trainer:
|
|
max_epochs: 50
|
|
callbacks:
|
|
- class_path: lightning.pytorch.callbacks.EarlyStopping
|
|
init_args:
|
|
patience: 2
|
|
monitor: pixel_AUROC
|
|
mode: max
|