use abs on early exit condition

shuklabhay · Sep 25, 2024 · ce694f5 · ce694f5
1 parent aeb63f3
commit ce694f5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/train.py b/src/train.py
@@ -289,8 +289,8 @@ def training_loop(train_loader, val_loader):
                 )
 
         # Early exit/saving
-        if val_w_dist < best_val_w_dist:
-            best_val_w_dist = val_w_dist
+        if np.abs(val_w_dist) < best_val_w_dist:
+            best_val_w_dist = np.abs(val_w_dist)
             epochs_no_improve = 0
             save_model(generator)
         else: