Update train_sagan.py

Fix the error
htyj123 · Apr 25, 2019 · 22455da · 22455da
1 parent 5fe132f
commit 22455da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_sagan.py b/train_sagan.py
@@ -74,7 +74,7 @@ def validate(netG, netD, GANLoss, ReconLoss, DLoss, optG, optD, dataloader, epoc
         imgs = (imgs / 127.5 - 1)
         # mask is 1 on masked region
         # forward
-        coarse_imgs, recon_imgs, attention = netG(imgs, masks)
+        coarse_imgs, recon_imgs = netG(imgs, masks)
 
         complete_imgs = recon_imgs * masks + imgs * (1 - masks)