fix: criterionIdt is not used by cut

jolibrain · Feb 3, 2022 · 3665410 · 3665410
1 parent 31562b9
commit 3665410
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/models/cut_model.py b/models/cut_model.py
@@ -116,7 +116,6 @@ def __init__(self, opt,rank):
             for nce_layer in self.nce_layers:
                 self.criterionNCE.append(PatchNCELoss(opt).to(self.device))
 
-            self.criterionIdt = torch.nn.L1Loss().to(self.device)
             self.optimizer_G = torch.optim.Adam(self.netG.parameters(), lr=opt.lr, betas=(opt.beta1, opt.beta2))
             if opt.netD_global== "none":
                 self.optimizer_D = torch.optim.Adam(self.netD.parameters(), lr=opt.lr, betas=(opt.beta1, opt.beta2))