Skip to content

N
neural_filters
Commit f420f932 authored by Bastian SCHNELL's avatar Bastian SCHNELL
Browse files
Options

Parameter still required to register tensor.

parent ea285b4c
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
neural_filters/neural_filter.py
View file @ f420f932
......@@ -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
View file @ f420f932
......@@ -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