提交 ca367a80 编写于 作者: D dangqingqing

follow comments

上级 ffd045a0
......@@ -74,9 +74,7 @@ class Topology(object):
if isinstance(layer, v2_layer.DataLayerV2):
data_layers[layer.name] = layer
if not isinstance(layer, collections.Sequence):
for parent_layer in layer.__parent_layers__.values():
find_data_layer(parent_layer)
else:
layer = [layer]
for each_l in layer:
for parent_layer in each_l.__parent_layers__.values():
find_data_layer(parent_layer)
......@@ -93,8 +91,8 @@ class Topology(object):
"""
data_types_lists = []
for each in self.__model_config__.input_layer_names:
data_layers = self.data_layers()
for each in self.__model_config__.input_layer_names:
data_types_lists.append((each, data_layers[each].type))
return data_types_lists
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册