Update dcgan_faces_tutorial.py

Update labels to be floats to work around torch.full inference change.
mthrok · Jul 28, 2020 · 11569e0 · 11569e0
1 parent afede4c
commit 11569e0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/beginner_source/dcgan_faces_tutorial.py b/beginner_source/dcgan_faces_tutorial.py
@@ -493,8 +493,8 @@ def forward(self, input):
 fixed_noise = torch.randn(64, nz, 1, 1, device=device)
 
 # Establish convention for real and fake labels during training
-real_label = 1
-fake_label = 0
+real_label = 1.
+fake_label = 0.
 
 # Setup Adam optimizers for both G and D
 optimizerD = optim.Adam(netD.parameters(), lr=lr, betas=(beta1, 0.999))