wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-07-31 12:40:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

shape has to be a tuple

Browse Source
This commit is contained in:
Kashif Rasul
2020-01-13 19:52:47 +01:00
parent ca2918b02e
commit 8cb33f20c6
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/modules/flows.py
+2 -2
View File
@@ -196,9 +196,9 @@ class RealNVP(nn.Module):
def sample(self, sample_shape=torch.Size()):
if self.cond is not None:
shape = self.cond.shape[:-1] + (self.input_size)
shape = self.cond.shape[:-1] + (self.input_size,)
else:
shape = sample_shape + (self.input_size)
shape = sample_shape + (self.input_size,)
# if len(sample_shape) > 0:
# shape = sample_shape + (self.input_size)