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

revert

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-01-15 13:24:57 +01:00
parent 888139203a
commit c1970eae9b
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
@@ -171,13 +171,13 @@ class RealNVP(nn.Module):
def forward(self, x, cond):
if self.scale is not None:
x *= self.scale
x /= self.scale
return self.net(x, cond)
def inverse(self, u, cond):
x, log_abs_det_jacobian = self.net.inverse(u, cond)
if self.scale is not None:
x /= self.scale
x *= self.scale
return x, log_abs_det_jacobian
def log_prob(self, x, cond):