Update rotatespade_model.py

Hangz-nju-cuhk · May 1, 2020 · a2cbd23 · a2cbd23
1 parent d1a9232
commit a2cbd23
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/models/rotatespade_model.py b/models/rotatespade_model.py
@@ -33,8 +33,6 @@ def __init__(self, opt):
             self.criterionFeat = torch.nn.L1Loss()
             if not opt.no_vgg_loss:
                 self.criterionVGG = networks.VGGLoss(self.opt)
-            if opt.use_vae:
-                self.KLDLoss = networks.KLDLoss()
 
     def landmark_68_to_5(self, t68):
         le = t68[36:42, :].mean(axis=0, keepdims=True)