add scale variable

2026-07-21 12:50:26 +08:00 · 2020-01-17 13:14:31 +01:00
parent e472f57c1c
commit ea2466f8a0
1 changed files with 2 additions and 1 deletions
@@ -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