Reworked the reset weights/biases function

ArtificialIntelligenceToolkit · Sep 12, 2018 · f34dc71 · f34dc71
1 parent 26d667f
commit f34dc71
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/conx/_version.py b/conx/_version.py
@@ -17,6 +17,6 @@
 # Foundation, Inc., 51 Franklin Street, Fifth Floor,
 # Boston, MA 02110-1301  USA
 
-__version__ = "3.7.4"
+__version__ = "3.7.5"
 VERSION = tuple([(int(v) if v.isdigit() else v)
                  for v in __version__.split(".")])
diff --git a/conx/network.py b/conx/network.py
@@ -883,10 +883,10 @@ def reset_weights(self):
         """
         session = K.get_session()
         for layer in self.model.layers:
-            if hasattr(layer, 'kernel_initializer'):
+            if hasattr(layer, 'kernel'):
                 layer.kernel.initializer.run(session=session)
-            if hasattr(layer, 'bias_initializer'):
-                layer.bias.initializer.run(session=session)
+            elif hasattr(layer, 'cell'):
+                layer.cell.kernel.initializer.run(session=session)
 
     def test(self, batch_size=32, show=False, tolerance=None, force=False,
              show_inputs=True, show_outputs=True,