From f420f9320cb46031190b159c98b06b7654e2d344 Mon Sep 17 00:00:00 2001 From: Bastian SCHNELL Date: Fri, 23 Aug 2019 14:19:41 +0200 Subject: [PATCH] Parameter still required to register tensor. --- neural_filters/neural_filter.py | 2 +- neural_filters/neural_filter_2CC.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/neural_filters/neural_filter.py b/neural_filters/neural_filter.py index 4ee0167..e80b6c7 100644 --- a/neural_filters/neural_filter.py +++ b/neural_filters/neural_filter.py @@ -34,7 +34,7 @@ class NeuralFilter(torch.nn.Module): self.hidden_size = hidden_size - self.register_parameter('bias_forget', torch.Tensor(hidden_size)) + self.register_parameter('bias_forget', Parameter(torch.Tensor(hidden_size))) self.reset_parameters() diff --git a/neural_filters/neural_filter_2CC.py b/neural_filters/neural_filter_2CC.py index 03f49ab..efbb4aa 100644 --- a/neural_filters/neural_filter_2CC.py +++ b/neural_filters/neural_filter_2CC.py @@ -35,8 +35,8 @@ class NeuralFilter2CC(torch.nn.Module): self.hidden_size = hidden_size - self.register_parameter('bias_theta', torch.Tensor(hidden_size)) - self.register_parameter('bias_modulus', torch.Tensor(hidden_size)) + self.register_parameter('bias_theta', Parameter(torch.Tensor(hidden_size))) + self.register_parameter('bias_modulus', Parameter(torch.Tensor(hidden_size))) self.reset_parameters() -- 2.21.0