From 9ac375a89db68d52ac74f5aa780aa1851c3946a3 Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Thu, 4 Jun 2020 10:34:58 +0200 Subject: [PATCH] log_abs_det doesnt change int he forward --- pts/modules/flows.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/pts/modules/flows.py b/pts/modules/flows.py index b00ab58..52f1e09 100644 --- a/pts/modules/flows.py +++ b/pts/modules/flows.py @@ -327,8 +327,6 @@ class Flow(nn.Module): 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):