changed to gen_steps to 5

caffeine-coder1 · caffeine-coder1 · commit 0ede07a9598b · 2021-06-18T19:17:23.000+05:30
diff --git a/GAN/DCGAN/training.py b/GAN/DCGAN/training.py
@@ -23,7 +23,7 @@ def training(opt):
     work_device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     FEATURE_D = 128
     Z_DIM = 100
-    GEN_TRAIN_STEPS = 2
+    GEN_TRAIN_STEPS = 5
     BATCH_SIZE = opt.batch_size
 
     if opt.logs:
