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.
24 lines
442 B
YAML
24 lines
442 B
YAML
model:
|
|
class_path: anomalib.models.Ganomaly
|
|
init_args:
|
|
batch_size: 32
|
|
n_features: 64
|
|
latent_vec_size: 100
|
|
extra_layers: 0
|
|
add_final_conv_layer: true
|
|
wadv: 1
|
|
wcon: 50
|
|
wenc: 1
|
|
lr: 0.0002
|
|
beta1: 0.5
|
|
beta2: 0.999
|
|
|
|
trainer:
|
|
max_epochs: 100
|
|
callbacks:
|
|
- class_path: lightning.pytorch.callbacks.EarlyStopping
|
|
init_args:
|
|
patience: 3
|
|
monitor: image_AUROC
|
|
mode: max
|