Update STFPM config (#3935)

ashwinvaidya17 · web-flow · commit c2705df953df · 2024-09-06T16:28:30.000+09:00
diff --git a/src/otx/recipe/anomaly_classification/stfpm.yaml b/src/otx/recipe/anomaly_classification/stfpm.yaml
@@ -19,6 +19,10 @@ overrides:
     - class_path: lightning.pytorch.callbacks.EarlyStopping
       init_args:
         patience: 5
+        mode: max
+        monitor: pixel_F1Score
     - class_path: otx.algo.callbacks.adaptive_train_scheduling.AdaptiveTrainScheduling
       init_args:
         max_interval: 1
+  gradient_clip_val: 0
+  num_sanity_val_steps: 0
diff --git a/src/otx/recipe/anomaly_detection/stfpm.yaml b/src/otx/recipe/anomaly_detection/stfpm.yaml
@@ -24,9 +24,13 @@ overrides:
     - class_path: lightning.pytorch.callbacks.EarlyStopping
       init_args:
         patience: 5
+        mode: max
+        monitor: pixel_F1Score
     - class_path: otx.algo.callbacks.adaptive_train_scheduling.AdaptiveTrainScheduling
       init_args:
         max_interval: 1
+  gradient_clip_val: 0
+  num_sanity_val_steps: 0
 
   data:
     task: ANOMALY_DETECTION
diff --git a/src/otx/recipe/anomaly_segmentation/stfpm.yaml b/src/otx/recipe/anomaly_segmentation/stfpm.yaml
@@ -19,9 +19,13 @@ overrides:
     - class_path: lightning.pytorch.callbacks.EarlyStopping
       init_args:
         patience: 5
+        mode: max
+        monitor: pixel_F1Score
     - class_path: otx.algo.callbacks.adaptive_train_scheduling.AdaptiveTrainScheduling
       init_args:
         max_interval: 1
+  gradient_clip_val: 0
+  num_sanity_val_steps: 0
 
   data:
     task: ANOMALY_SEGMENTATION
