changed the learning rate

caffeine-coder1 · caffeine-coder1 · commit 7f308091dd12 · 2021-06-30T16:04:43.000+05:30
diff --git a/GAN/WGAN/training.py b/GAN/WGAN/training.py
@@ -27,7 +27,7 @@ def training(opt):
 
     # ~~~~~~~~~~~~~~~~~~~ as per WGAN paper ~~~~~~~~~~~~~~~~~~~ #
 
-    lr = opt.lr if opt.lr else 5e-5
+    lr = opt.lr
     CRITIC_TRAIN_STEPS = 5
     WEIGHT_CLIP = 0.01
 
@@ -203,7 +203,7 @@ def training(opt):
                         help='number of epochs to train')
     parser.add_argument('--batch-size', type=int, default=128,
                         help='total batch size for all GPUs')
-    parser.add_argument('--lr', type=float, default=2e-4,
+    parser.add_argument('--lr', type=float, default=5e-5,
                         help='learning rate to use')
 
     parser.add_argument('--resume', type=bool, default=True,
