From ea2466f8a0aa6f381c100a5dc272063bc9cde686 Mon Sep 17 00:00:00 2001
From: "Dr. Kashif Rasul" <kashif.rasul@gmail.com>
Date: Fri, 17 Jan 2020 13:14:31 +0100
Subject: [PATCH] add scale variable

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

diff --git a/pts/modules/flows.py b/pts/modules/flows.py
index 483014d..b74cf8d 100644
--- a/pts/modules/flows.py
+++ b/pts/modules/flows.py
@@ -371,7 +371,8 @@ class MAF(nn.Module):
         # base distribution for calculation of log prob under the model
         self.register_buffer("base_dist_mean", torch.zeros(input_size))
         self.register_buffer("base_dist_var", torch.ones(input_size))
-
+        self.__scale = None
+        
         # construct model
         modules = []
         self.input_degrees = None