changed the testing code for models.

caffeine-coder1 · caffeine-coder1 · commit 92d0084c0821 · 2021-06-18T13:19:30.000+05:30
diff --git a/GAN/DCGAN/model.py b/GAN/DCGAN/model.py
@@ -69,9 +69,9 @@ def forward(self, x):
 
 
 if __name__ == "__main__":
-    disc = Discriminator(3, 128)
-    gen = Faker(100, 3, 128)
+    disc = Discriminator(1, 128)
+    gen = Faker(100, 1, 128)
 
-    summary(disc, (3, 64, 64))
+    summary(disc, (1, 64, 64))
 
     summary(gen, (100, 1, 1))
