Register parameters for dict saving/loading.

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

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

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',