From 117d2ccd9d7b2e517e8b345735edd4dd2b5f7111 Mon Sep 17 00:00:00 2001 From: rajeevyasarla Date: Wed, 31 Jul 2019 11:38:08 -0400 Subject: [PATCH] UMSN files --- train_face_deblur.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_face_deblur.py b/train_face_deblur.py index db294dd..e781dc6 100644 --- a/train_face_deblur.py +++ b/train_face_deblur.py @@ -356,7 +356,7 @@ for epoch in range(opt.niter): #x1 = xmask1*class_msk1*x_hat+(1-xmask1)*class_msk1*target #smaps_hat,smaps64_hat = netS(x_hat1,x_hat64) - if epoch>4 or (epoch<4 and epoch%2 == 0): + if epoch>4 or (epoch<4 and ganIterations%10 ~= 9): with torch.no_grad(): smaps,smaps64 = netS(target,target_256) L_img_ = 0.33*criterionCAE(x_hat64, target_256) #+ 0.5*criterionCAE(smaps_hat, smaps)