Commit ea285b4c authored by François MARELLI's avatar François MARELLI

Register parameters for dict saving/loading.

parent 86fe8641
...@@ -34,7 +34,7 @@ class NeuralFilter(torch.nn.Module): ...@@ -34,7 +34,7 @@ class NeuralFilter(torch.nn.Module):
self.hidden_size = hidden_size 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() self.reset_parameters()
......
...@@ -35,8 +35,8 @@ class NeuralFilter2CC(torch.nn.Module): ...@@ -35,8 +35,8 @@ class NeuralFilter2CC(torch.nn.Module):
self.hidden_size = hidden_size self.hidden_size = hidden_size
self.bias_theta = Parameter(torch.Tensor(hidden_size)) self.register_parameter('bias_theta', torch.Tensor(hidden_size))
self.bias_modulus = Parameter(torch.Tensor(hidden_size)) self.register_parameter('bias_modulus', torch.Tensor(hidden_size))
self.reset_parameters() self.reset_parameters()
......
...@@ -2,7 +2,7 @@ from setuptools import setup ...@@ -2,7 +2,7 @@ from setuptools import setup
setup( setup(
name='neural-filters', name='neural-filters',
version='0.1', version='0.2',
description='Linear filters for neural networks in pyTorch', description='Linear filters for neural networks in pyTorch',
author='François Marelli (Idiap research institute)', author='François Marelli (Idiap research institute)',
author_email='francois.marelli@idiap.ch', author_email='francois.marelli@idiap.ch',
......
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