diff --git a/src/otx/recipe/anomaly_classification/stfpm.yaml b/src/otx/recipe/anomaly_classification/stfpm.yaml index 47f16e5429d..3364156dcb0 100644 --- 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 index 6253bc6cfb7..c89d3f6db91 100644 --- 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 index eae9d551f7b..22c2673fac1 100644 --- 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