fix shape

2026-07-14 11:17:47 +08:00 · 2020-01-13 20:14:20 +01:00
parent 0a8e9fdec8
commit 5c0e84c2fd
1 changed files with 2 additions and 2 deletions
@@ -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] 
        else:
-            shape = sample_shape + (self.input_size,)
+            shape = sample_shape
        # if len(sample_shape) > 0:
        #     shape = sample_shape + (self.input_size)