From 633cf580488536d653275725415d6311e0a0d7db Mon Sep 17 00:00:00 2001 From: rajeevyasarla Date: Tue, 10 Sep 2019 11:10:31 -0400 Subject: [PATCH] Update train_face_deblur.py --- 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 cab574c..ba23490 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>0: + if epoch>-1: 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)