wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-06-27 18:06:19 +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
2020-06-04 10:34:58 +02:00
parent 6efcd6c5b9
commit 9ac375a89d
1 changed files with 0 additions and 2 deletions
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):