Merge pull request #736 from zhaoming0/fixbug-735

fix #735
mil-tokyo · Jan 19, 2018 · e04b274 · e04b274
2 parents 7a901af + f423ad8
commit e04b274
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/graph_transpiler/webdnn/frontend/keras/converter.py b/src/graph_transpiler/webdnn/frontend/keras/converter.py
@@ -124,8 +124,8 @@ def _convert_fallback(self, model: "keras.models.Model") -> Graph:
             if not Placeholder.check_resolved(v.shape[0]):
                 v.shape[0].value = self._batch_size
 
-        for depth in sorted(list(model.nodes_by_depth.keys()), reverse=True):
-            for node in model.nodes_by_depth[depth]:
+        for depth in sorted(list(model._nodes_by_depth.keys()), reverse=True):
+            for node in model._nodes_by_depth[depth]:
                 self._convert_operator(node.outbound_layer)
 
                 # Check that all output tensors from current layer are converted into WebDNN Variable