Set stfpm precision to 32 (#3484)

openvinotoolkit · May 10, 2024 · faed8f7 · faed8f7
1 parent 377de90
commit faed8f7
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/src/otx/recipe/anomaly_classification/stfpm.yaml b/src/otx/recipe/anomaly_classification/stfpm.yaml
@@ -20,6 +20,7 @@ callback_monitor: image_F1Score
 
 data: ../_base_/data/torchvision_base.yaml
 overrides:
+  precision: 32
   max_epochs: 100
   callbacks:
     - class_path: lightning.pytorch.callbacks.EarlyStopping

diff --git a/src/otx/recipe/anomaly_detection/stfpm.yaml b/src/otx/recipe/anomaly_detection/stfpm.yaml
@@ -20,6 +20,7 @@ callback_monitor: image_F1Score
 
 data: ../_base_/data/torchvision_base.yaml
 overrides:
+  precision: 32
   max_epochs: 100
   callbacks:
     - class_path: lightning.pytorch.callbacks.EarlyStopping

diff --git a/src/otx/recipe/anomaly_segmentation/stfpm.yaml b/src/otx/recipe/anomaly_segmentation/stfpm.yaml
@@ -20,6 +20,7 @@ callback_monitor: image_F1Score
 
 data: ../_base_/data/torchvision_base.yaml
 overrides:
+  precision: 32
   max_epochs: 100
   callbacks:
     - class_path: lightning.pytorch.callbacks.EarlyStopping