From f8bf093fee76cfead70e445b9add0830505b12d3 Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Mon, 13 Jan 2020 20:15:31 +0100 Subject: [PATCH] typo --- pts/modules/flows.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pts/modules/flows.py b/pts/modules/flows.py index 42fde26..f1b430d 100644 --- a/pts/modules/flows.py +++ b/pts/modules/flows.py @@ -187,7 +187,7 @@ class RealNVP(nn.Module): def inverse(self, u): x, log_abs_det_jacobian = self.net.inverse(u, self.cond) if self.scale is not None: - x *= scale + x *= self.scale return x, log_abs_det_jacobian def log_prob(self, x):