From faed8f7dca8dcbad2af1d52ec99d998aad4fe713 Mon Sep 17 00:00:00 2001 From: Ashwin Vaidya Date: Fri, 10 May 2024 13:45:29 +0200 Subject: [PATCH] Set stfpm precision to 32 (#3484) --- src/otx/recipe/anomaly_classification/stfpm.yaml | 1 + src/otx/recipe/anomaly_detection/stfpm.yaml | 1 + src/otx/recipe/anomaly_segmentation/stfpm.yaml | 1 + 3 files changed, 3 insertions(+) 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