changes to fix issue #8381 (#8391)

keras-team · Nov 5, 2017 · bbcf0e6 · bbcf0e6
1 parent 3292aa5
commit bbcf0e6
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/keras/engine/training.py b/keras/engine/training.py
@@ -22,7 +22,6 @@
 from .. import losses
 from .. import metrics as metrics_module
 from ..utils.generic_utils import Progbar
-from ..utils.layer_utils import count_params
 from .. import callbacks as cbks
 from ..legacy import interfaces
 
@@ -967,8 +966,8 @@ def _check_trainable_weights_consistency(self):
         if not hasattr(self, '_collected_trainable_weights'):
             return
 
-        if (count_params(self.trainable_weights) !=
-                count_params(self._collected_trainable_weights)):
+        if (len(self.trainable_weights) !=
+                len(self._collected_trainable_weights)):
             warnings.warn(UserWarning(
                 'Discrepancy between trainable weights and collected trainable'
                 ' weights, did you set `model.trainable` without calling'