Fix issue with cascade of Merge layers

keras-team · Jun 13, 2016 · c9aee41 · c9aee41
1 parent 3b83a1b
commit c9aee41
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/keras/engine/training.py b/keras/engine/training.py
@@ -235,7 +235,10 @@ def collect_trainable_weights(layer):
     if not trainable:
         return []
     weights = []
-    if layer.__class__.__name__ in ['Sequential', 'Model']:
+    if layer.__class__.__name__ == 'Sequential':
+        for sublayer in layer.flattened_layers:
+            weights += collect_trainable_weights(sublayer)
+    elif layer.__class__.__name__ == 'Model':
         for sublayer in layer.layers:
             weights += collect_trainable_weights(sublayer)
     elif layer.__class__.__name__ == 'Graph':