setting opt.identity other than 1 has an effect of scaling the weight…

… of the identity mapping loss. For example, if the weight of the identity loss should be 10 times smaller than the weight of the reconstruction loss, please set opt.identity = 0.1
crabsbeard · Apr 3, 2017 · 3fb35c1 · 3fb35c1
1 parent f7801e1
commit 3fb35c1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/models/cycle_gan_model.lua b/models/cycle_gan_model.lua
@@ -166,10 +166,10 @@ function CycleGANModel:fGx_basic(x, gradParams, netG, netD, netE, real, real2, r
   -- G should be identity if real2 is fed.
   local errI = nil
   local identity = nil
-  if opt.identity == 1 then
+  if opt.identity > 0 then
     identity = netG:forward(real2):clone()
-    errI = self.criterionRec:forward(identity, real2)*lambda2
-    local didentity_loss_do = self.criterionRec:backward(identity, real2):mul(lambda2)
+    errI = self.criterionRec:forward(identity, real2)*lambda2*opt.identity
+    local didentity_loss_do = self.criterionRec:backward(identity, real2):mul(lambda2):mul(opt.identity)
     netG:backward(real2, didentity_loss_do)
   end