Commit f420f932 authored by Bastian SCHNELL's avatar Bastian SCHNELL

Parameter still required to register tensor.

parent ea285b4c
......@@ -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()
......
......@@ -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()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment