Removing extra flatten (#809)

Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
Lightning-Universe · May 4, 2022 · 72614f0 · 72614f0
1 parent 3842cf3
commit 72614f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pl_bolts/models/gans/basic/basic_gan_module.py b/pl_bolts/models/gans/basic/basic_gan_module.py
@@ -92,7 +92,7 @@ def generator_loss(self, x):
     def discriminator_loss(self, x):
         # train discriminator on real
         b = x.size(0)
-        x_real = x.view(b, -1)
+        x_real = x
         y_real = torch.ones(b, 1, device=self.device)
 
         # calculate real score