From ea285b4ce9cfbac3df6a1e1343c818a6df4dc586 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fran=C3=A7ois=20MARELLI?= Date: Fri, 23 Aug 2019 14:09:36 +0200 Subject: [PATCH] Register parameters for dict saving/loading. --- neural_filters/neural_filter.py | 2 +- neural_filters/neural_filter_2CC.py | 4 ++-- setup.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/neural_filters/neural_filter.py b/neural_filters/neural_filter.py index a7550c2..4ee0167 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.bias_forget = Parameter(torch.Tensor(hidden_size)) + self.register_parameter('bias_forget', torch.Tensor(hidden_size)) self.reset_parameters() diff --git a/neural_filters/neural_filter_2CC.py b/neural_filters/neural_filter_2CC.py index 6e2e294..03f49ab 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.bias_theta = Parameter(torch.Tensor(hidden_size)) - self.bias_modulus = Parameter(torch.Tensor(hidden_size)) + self.register_parameter('bias_theta', torch.Tensor(hidden_size)) + self.register_parameter('bias_modulus', torch.Tensor(hidden_size)) self.reset_parameters() diff --git a/setup.py b/setup.py index d37047e..3022b39 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ from setuptools import setup setup( name='neural-filters', - version='0.1', + version='0.2', description='Linear filters for neural networks in pyTorch', author='François Marelli (Idiap research institute)', author_email='francois.marelli@idiap.ch', -- 2.21.0