Lamda_0.4, FM_8

code/model_GAN.py

@@ -112,7 +112,7 @@ class Discriminator(nn.Module):
         self.learning_rate = learning_rate
         self.name = name
         self.d_model = DCGANDiscriminator(
-        feature_maps = 128,
+        feature_maps = 8,
         image_channels = 1
         )
 
@@ -140,7 +140,7 @@ class GAN(pl.LightningModule):
         dict_args = vars(args)
         return cls(**dict_args)
 
-    def __init__(self, learning_rate=1e-3, name='model', depth=3, lamda_G = 0.2, **kwargs):
+    def __init__(self, learning_rate=1e-3, name='model', depth=3, lamda_G = 0.4, **kwargs):
         super().__init__()
         self.save_hyperparameters()
         self.learning_rate = learning_rate