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

log_abs_det doesnt change int he forward

Browse Source
This commit is contained in:
Dr. Kashif Rasul committed 2020-06-04 10:34:58 +02:00
1 parent 6efcd6c5b9
commit 9ac375a89d
1 file changed
-2
Download Patch File Download Diff File Expand all files Collapse all files
pts/modules/flows.py
-2
View File
@@ -327,8 +327,6 @@ class Flow(nn.Module):
if self.scale is not None:
x /= self.scale
u, log_abs_det_jacobian = self.net(x, cond)
if self.scale is not None:
log_abs_det_jacobian -= torch.log(torch.abs(self.scale))
return u, log_abs_det_jacobian
def inverse(self, u, cond):