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

fix comparison

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-01-13 16:59:07 +01:00
parent 7cea995d10
commit ca2918b02e
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/modules/flows.py
+6 -4
View File
@@ -180,13 +180,13 @@ class RealNVP(nn.Module):
self.__scale = scale
def forward(self, x):
if self.scale:
if self.scale is not None:
x /= self.scale
return self.net(x, self.cond)
def inverse(self, u):
x, log_abs_det_jacobian = self.net.inverse(u, self.cond)
if self.scale:
if self.scale is not None:
x *= scale
return x, log_abs_det_jacobian
@@ -195,10 +195,12 @@ class RealNVP(nn.Module):
return torch.sum(self.base_dist.log_prob(u) + sum_log_abs_det_jacobians, dim=-1)
def sample(self, sample_shape=torch.Size()):
if self.cond:
if self.cond is not None:
shape = self.cond.shape[:-1] + (self.input_size)
if sample_shape is not None:
else:
shape = sample_shape + (self.input_size)
# if len(sample_shape) > 0:
# shape = sample_shape + (self.input_size)
u = self.base_dist.sample(shape)
sample, _ = self.inverse(u)