diff --git a/neural_filters/neural_filter.py b/neural_filters/neural_filter.py index 4ee01679b293be73f5bf902c9ed810e4c124b247..e80b6c7fe3bac3d7fd15904688cd5d3674b82bb9 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 03f49ab36db1201614a3b979c5be63c708e14aa2..efbb4aa6e3f81fc83ad95d6987bf1b362b9a6364 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()