Merge pull request #9 from 1nadequacy/master

F.tanh -> torch.tanh
ikostrikov · Mar 17, 2019 · 5520ebe · 5520ebe
2 parents e679e05 + 3bd1fd1
commit 5520ebe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flows.py b/flows.py
@@ -107,7 +107,7 @@ def forward(self, inputs, cond_inputs=None, mode='direct'):
             a = self.t_trunk(h)
 
             if self.pre_exp_tanh:
-                a = F.tanh(a)
+                a = torch.tanh(a)
 
             u = (inputs - m) * torch.exp(-a)
             return u, -a.sum(-1, keepdim=True)
@@ -122,7 +122,7 @@ def forward(self, inputs, cond_inputs=None, mode='direct'):
                 a = self.t_trunk(h)
 
                 if self.pre_exp_tanh:
-                    a = F.tanh(a)
+                    a = torch.tanh(a)
 
                 x[:, i_col] = inputs[:, i_col] * torch.exp(
                     a[:, i_col]) + m[:, i_col]