Update train_face_deblur.py

2020-05-17 19:26:39 -04:00
parent d5272892e3
commit a50a7b9cb2
1 changed files with 1 additions and 1 deletions
@@ -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"]