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
2020-05-17 19:26:39 -04:00
committed by GitHub
parent d5272892e3
commit a50a7b9cb2
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
@@ -80,7 +80,7 @@ kernels = kernels.transpose([0,2,1])
vgg = Vgg16() vgg = Vgg16()
utils.init_vgg16('./models/') #utils.init_vgg16('./models/')
state_dict_g = torch.load('VGG_FACE.pth') state_dict_g = torch.load('VGG_FACE.pth')
new_state_dict_g = {} new_state_dict_g = {}
new_state_dict_g["conv1_1.weight"]= state_dict_g["0.weight"] new_state_dict_g["conv1_1.weight"]= state_dict_g["0.weight"]