From 6bc8de7edba16cbb815d6c2ed9c34faf4e3cf3e3 Mon Sep 17 00:00:00 2001
From: Kashif Rasul <kashif.rasul@gmail.com>
Date: Thu, 5 Mar 2020 15:37:45 +0100
Subject: [PATCH] call __init__ before registering buffers

---
 pts/modules/flows.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pts/modules/flows.py b/pts/modules/flows.py
index 582e8df..5f90f27 100644
--- a/pts/modules/flows.py
+++ b/pts/modules/flows.py
@@ -304,6 +304,7 @@ class MADE(nn.Module):
 
 class Flow(nn.Module):
     def __init__(self, input_size):
+        super().__init__()
         self.__scale = None
         self.net = None