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

cleanup

Browse Source
This commit is contained in:
Kashif Rasul
2020-01-13 20:31:05 +01:00
parent f8bf093fee
commit c429003932
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/modules/flows.py
-3
View File
@@ -144,7 +144,6 @@ class RealNVP(nn.Module):
self.__cond = None
self.__scale = None
self.input_size = input_size
# construct model
modules = []
@@ -199,8 +198,6 @@ class RealNVP(nn.Module):
shape = self.cond.shape[:-1]
else:
shape = sample_shape
# if len(sample_shape) > 0:
# shape = sample_shape + (self.input_size)
u = self.base_dist.sample(shape)
sample, _ = self.inverse(u)