This commit is contained in:
Dr. Kashif Rasul
2020-03-31 11:04:56 +02:00
parent cb012405bf
commit 192f9d3991
+1 -1
View File
@@ -275,7 +275,7 @@ class MADE(nn.Module):
@property
def base_dist(self):
return D.Normal(self.base_dist_mean, self.base_dist_var)
return Normal(self.base_dist_mean, self.base_dist_var)
def forward(self, x, y=None):
# MAF eq 4 -- return mean and log std