Update train_face_deblur.py

2020-05-17 19:29:33 -04:00
parent a50a7b9cb2
commit 60516e9633
1 changed files with 1 additions and 5 deletions
@@ -248,11 +248,7 @@ target_256 = Variable(target_256)
 input_256 = Variable(input_256)
 ato = Variable(ato)
-# Initialize VGG-16
+
 vgg = Vgg16()
 utils.init_vgg16('./models/')
 vgg.load_state_dict(torch.load(os.path.join('./models/', "vgg16.weight")))
 vgg.cuda()
 label_d = Variable(label_d.cuda())