From 22455dae24c42b58ac3fe4803c35487fe56d56dd Mon Sep 17 00:00:00 2001
From: avalonstrel <461202177@qq.com>
Date: Thu, 25 Apr 2019 16:09:45 +0800
Subject: [PATCH] Update train_sagan.py

Fix the error
---
 train_sagan.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train_sagan.py b/train_sagan.py
index e0ca583..59a54c5 100644
--- 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)