call __init__ before registering buffers

2026-07-15 11:25:33 +08:00 · 2020-03-05 15:37:45 +01:00
parent 5bb0d7d6b6
commit 6bc8de7edb
1 changed files with 1 additions and 0 deletions
@@ -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