From 32d58f9d7e9e1d98951987f71090c19fc36b3751 Mon Sep 17 00:00:00 2001
From: Stefan Klut <stefanklut12@gmail.com>
Date: Wed, 20 Mar 2024 18:22:33 +0100
Subject: [PATCH] auto turn on amp

---
 configs/segmentation/baseline/baseline_general.yaml | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/configs/segmentation/baseline/baseline_general.yaml b/configs/segmentation/baseline/baseline_general.yaml
index d5ecd9e..4404bfe 100644
--- a/configs/segmentation/baseline/baseline_general.yaml
+++ b/configs/segmentation/baseline/baseline_general.yaml
@@ -102,6 +102,9 @@ MODEL:
   SEM_SEG_HEAD:
     NUM_CLASSES: 2
 
+  AMP_TRAIN:
+    ENABLED: True
+
   WEIGHTS:
 
 TRAIN: