Parameter still required to register tensor.

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()
 

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()