From c429003932d7c265977e69e62ec13547c4bac34c Mon Sep 17 00:00:00 2001
From: Kashif Rasul <kashif.rasul@gmail.com>
Date: Mon, 13 Jan 2020 20:31:05 +0100
Subject: [PATCH] cleanup

---
 pts/modules/flows.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/pts/modules/flows.py b/pts/modules/flows.py
index f1b430d..01c8bfe 100644
--- a/pts/modules/flows.py
+++ b/pts/modules/flows.py
@@ -144,7 +144,6 @@ class RealNVP(nn.Module):
 
         self.__cond = None
         self.__scale = None
-        self.input_size = input_size
 
         # construct model
         modules = []
@@ -199,8 +198,6 @@ class RealNVP(nn.Module):
             shape = self.cond.shape[:-1] 
         else:
             shape = sample_shape
-        # if len(sample_shape) > 0:
-        #     shape = sample_shape + (self.input_size)
 
         u = self.base_dist.sample(shape)
         sample, _ = self.inverse(u)