Fix density

stefanonardo · May 3, 2019 · a75b4c6 · a75b4c6
1 parent 5ac1a5e
commit a75b4c6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 
 setup(name='pytorch-esn',
-      version='1.2.2',
+      version='1.2.3',
       packages=find_packages(),
       install_requires=[
           'torch',

diff --git a/torchesn/nn/reservoir.py b/torchesn/nn/reservoir.py
@@ -65,7 +65,7 @@ def reset_parameters(self):
                     zero_weights = torch.randperm(
                         int(self.hidden_size * self.hidden_size))
                     zero_weights = zero_weights[
-                                   :round(
+                                   :int(
                                        self.hidden_size * self.hidden_size * (
                                                    1 - self.density))]
                     w_hh[zero_weights] = 0