curated kick train

shuklabhay · Sep 27, 2024 · 43dce1f · 43dce1f
1 parent 785e53f
commit 43dce1f
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/outputs/StereoSampleGAN-CuratedKick.pth b/outputs/StereoSampleGAN-CuratedKick.pth
diff --git a/src/train.py b/src/train.py
@@ -250,7 +250,7 @@ def training_loop(train_loader, val_loader):
 
     best_val_w_dist = float("inf")  # Initialize
     epochs_no_improve = 0
-    patience = 5  # epochs
+    patience = 3  # epochs
     for epoch in range(N_EPOCHS):
         # Train
         train_g_loss, train_c_loss, train_w_dist = train_epoch(