some change to the G_

znxlwm · Dec 5, 2018 · 7b6d0e2 · 7b6d0e2
1 parent 0d183bb
commit 7b6d0e2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/WGAN.py b/WGAN.py
@@ -141,7 +141,9 @@ def train(self):
                 D_real = self.D(x_)
                 D_real_loss = -torch.mean(D_real)
 
-                G_ = self.G(z_)
+                with torch.no_grad():
+                    G_ = self.G(z_)
+
                 D_fake = self.D(G_)
                 D_fake_loss = torch.mean(D_fake)