GAN with PT_Lightning DCGANDiscriminator as Discriminator

code/model_GAN.py

@@ -111,13 +111,18 @@ class Discriminator(nn.Module):
         self.learning_rate = learning_rate
         self.name = name
         self.d_model = DCGANDiscriminator(
-        feature_maps = 1,
+        feature_maps = 8,
         image_channels = 1
         )
 
     def forward(self, x):
-        x = self.d_model(x)
-        return x
+        #x = self.d_model(x)
+        #self.disc(x)
+        x = self.d_model.disc(x)
+        #out = F.adaptive_avg_pool2d(x, (1, 1)).squeeze()
+        out = F.adaptive_avg_pool2d(x, (1, 1)).view((x.size(0), -1))
+
+        return out
 
 #GAN
 class GAN(pl.LightningModule):
@@ -140,7 +145,7 @@ class GAN(pl.LightningModule):
         self.learning_rate = learning_rate
         self.name = name
         self.generator = Generator()
-        self.discriminator = Discriminator_2((256,256))
+        self.discriminator = Discriminator((256,256))
         self.cyto_imag_generated = None
         self.lamda_G = lamda_G