change default parameter values for DRAEM (#495)

change default parameter values
openvinotoolkit · Aug 11, 2022 · 32ebbc3 · 32ebbc3
1 parent e4f8499
commit 32ebbc3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/anomalib/models/draem/config.yaml b/anomalib/models/draem/config.yaml
@@ -25,7 +25,7 @@ model:
   anomaly_source_path: null # optional, e.g. ./datasets/dtd
   lr: 0.0001
   early_stopping:
-    patience: 50
+    patience: 20
     metric: pixel_AUROC
     mode: max
   normalization_method: min_max # options: [none, min_max, cdf]
@@ -88,7 +88,7 @@ trainer:
   limit_val_batches: 1.0
   log_every_n_steps: 50
   log_gpu_memory: null
-  max_epochs: 100
+  max_epochs: 700
   max_steps: -1
   max_time: null
   min_epochs: null