From c1970eae9bac13e50e44a44631ca60b6222aa839 Mon Sep 17 00:00:00 2001
From: "Dr. Kashif Rasul" <kashif.rasul@gmail.com>
Date: Wed, 15 Jan 2020 13:24:57 +0100
Subject: [PATCH] revert

---
 pts/modules/flows.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pts/modules/flows.py b/pts/modules/flows.py
index 59980fc..44c087c 100644
--- a/pts/modules/flows.py
+++ b/pts/modules/flows.py
@@ -171,13 +171,13 @@ class RealNVP(nn.Module):
 
     def forward(self, x, cond):
         if self.scale is not None:
-            x *= self.scale
+            x /= self.scale
         return self.net(x, cond)
 
     def inverse(self, u, cond):
         x, log_abs_det_jacobian = self.net.inverse(u, cond)
         if self.scale is not None:
-            x /= self.scale
+            x *= self.scale
         return x, log_abs_det_jacobian
 
     def log_prob(self, x, cond):