From a50a7b9cb29a67a8fb8c3e5d89ef5a8d4d02648f Mon Sep 17 00:00:00 2001 From: rajeevyasarla Date: Sun, 17 May 2020 19:26:39 -0400 Subject: [PATCH] Update train_face_deblur.py --- train_face_deblur.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_face_deblur.py b/train_face_deblur.py index d62f4d7..c81a225 100644 --- 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"]