log_abs_det doesnt change int he forward

zalandoresearch · Jun 4, 2020 · 9ac375a · 9ac375a
1 parent 6efcd6c
commit 9ac375a
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/pts/modules/flows.py b/pts/modules/flows.py
@@ -327,8 +327,6 @@ def forward(self, x, cond):
         if self.scale is not None:
             x /= self.scale
         u, log_abs_det_jacobian = self.net(x, cond)
-        if self.scale is not None:
-            log_abs_det_jacobian -= torch.log(torch.abs(self.scale))
         return u, log_abs_det_jacobian
 
     def inverse(self, u, cond):