mirror of
https://github.com/wassname/pytorch-ts.git
synced 2026-08-03 13:10:21 +08:00
fix typo
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user