printing the hyper parameters

caffeine-coder1 · caffeine-coder1 · commit 97952e434057 · 2021-06-30T18:57:02.000+05:30
diff --git a/GAN/WGAN/training.py b/GAN/WGAN/training.py
@@ -25,13 +25,14 @@ def training(opt):
     FEATURE_D = 128
     Z_DIM = 100
     BATCH_SIZE = opt.batch_size
-
     # ~~~~~~~~~~~~~~~~~~~ as per WGAN paper ~~~~~~~~~~~~~~~~~~~ #
 
     lr = opt.lr
     CRITIC_TRAIN_STEPS = 5
     WEIGHT_CLIP = 0.01
 
+    print(f"Epochs: {EPOCHS}| lr: {lr}| batch size {BATCH_SIZE}" +
+          f"device: {work_device}")
     # ~~~~~~~~~~~ creating directories for weights ~~~~~~~~~~~ #
 
     if opt.logs:
