Update train_face_deblur.py

6 years ago · 633cf58048
parent 58e6924938
commit 633cf58048
1 changed files with 1 additions and 1 deletions
--- 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)