BATCH_SIZE decreased

Mafaz03 · Mafaz03 · commit 17962ad9dae8 · 2025-03-19T11:03:13.000+05:30
diff --git a/config.py b/config.py
@@ -10,7 +10,7 @@ def is_running_on_kaggle():
 TRAIN_DIR_A = "data/horse2zebra/trainA"
 TRAIN_DIR_B = "data/horse2zebra/trainB"
 VAL_DIR = "data/val"
-BATCH_SIZE = 5
+BATCH_SIZE = 1
 LEARNING_RATE = 1e-5
 LAMBDA_IDENTITY = 0.0
 LAMBDA_CYCLE = 10
