This commit is contained in:
Kashif Rasul
2020-01-13 20:15:31 +01:00
parent 5c0e84c2fd
commit f8bf093fee
+1 -1
View File
@@ -187,7 +187,7 @@ class RealNVP(nn.Module):
def inverse(self, u):
x, log_abs_det_jacobian = self.net.inverse(u, self.cond)
if self.scale is not None:
x *= scale
x *= self.scale
return x, log_abs_det_jacobian
def log_prob(self, x):