From 3c1ab0840078bd178f6aa5d8f695ff0556d74426 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?M=2E=20Fran=C3=A7ois?= Date: Mon, 22 Oct 2018 10:01:29 +0200 Subject: [PATCH] Tanh deprecated in pytorch --- neural_filters/neural_filter_2CC.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/neural_filters/neural_filter_2CC.py b/neural_filters/neural_filter_2CC.py index 1857ec4..00ef8ed 100644 --- a/neural_filters/neural_filter_2CC.py +++ b/neural_filters/neural_filter_2CC.py @@ -115,7 +115,7 @@ class NeuralFilter2CC(torch.nn.Module): def step(self, input_var, hidden, a=None, b=None): if a is None or b is None: modulus = torch.sigmoid(self.bias_modulus) - cosangle = F.tanh(self.bias_theta) + cosangle = torch.tanh(self.bias_theta) a = 2 * cosangle * modulus b = - modulus.pow(2) @@ -141,7 +141,7 @@ class NeuralFilter2CC(torch.nn.Module): # do not recompute this at each step to gain efficiency modulus = torch.sigmoid(self.bias_modulus) - cosangle = F.tanh(self.bias_theta) + cosangle = torch.tanh(self.bias_theta) a = 2 * cosangle * modulus b = - modulus.pow(2) @@ -155,7 +155,7 @@ class NeuralFilter2CC(torch.nn.Module): def print_param(self): modulus = torch.sigmoid(self.bias_modulus) - cosangle = F.tanh(self.bias_theta) + cosangle = torch.tanh(self.bias_theta) p1 = -2 * cosangle * modulus p2 = modulus.pow(2) print('{}\t{}'.format(p1.data[0], p2.data[0])) @@ -163,7 +163,7 @@ class NeuralFilter2CC(torch.nn.Module): @property def denominator(self): modulus = torch.sigmoid(self.bias_modulus) - cosangle = F.tanh(self.bias_theta) + cosangle = torch.tanh(self.bias_theta) p1 = -2 * cosangle * modulus p2 = modulus.pow(2) p1 = p1.detach().cpu().numpy() -- 2.21.0