fanwuyang/UMSN-Face-Deblurring
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update train_face_deblur.py

Browse Source
This commit is contained in:
rajeevyasarla
2019-09-10 11:10:31 -04:00
committed by GitHub
parent 58e6924938
commit 633cf58048
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train_face_deblur.py
+1 -1
View File
@@ -356,7 +356,7 @@ for epoch in range(opt.niter):
#x1 = xmask1*class_msk1*x_hat+(1-xmask1)*class_msk1*target #x1 = xmask1*class_msk1*x_hat+(1-xmask1)*class_msk1*target
#smaps_hat,smaps64_hat = netS(x_hat1,x_hat64) #smaps_hat,smaps64_hat = netS(x_hat1,x_hat64)
if epoch>0: if epoch>-1:
with torch.no_grad(): with torch.no_grad():
smaps,smaps64 = netS(target,target_256) smaps,smaps64 = netS(target,target_256)
L_img_ = 0.33*criterionCAE(x_hat64, target_256) #+ 0.5*criterionCAE(smaps_hat, smaps) L_img_ = 0.33*criterionCAE(x_hat64, target_256) #+ 0.5*criterionCAE(smaps_hat, smaps)