Update train_face_deblur.py

6 years ago · a50a7b9cb2
parent d5272892e3
commit a50a7b9cb2
1 changed files with 1 additions and 1 deletions
--- a/train_face_deblur.py
+++ b/train_face_deblur.py
@ -80,7 +80,7 @@ kernels = kernels.transpose([0,2,1])


 vgg = Vgg16()
-utils.init_vgg16('./models/')
+#utils.init_vgg16('./models/')
 state_dict_g = torch.load('VGG_FACE.pth')
 new_state_dict_g = {}
 new_state_dict_g["conv1_1.weight"]= state_dict_g["0.weight"]