diff --git a/python/paddle/v2/topology.py b/python/paddle/v2/topology.py index 5109dc58fe5128ab4144224f76190c5cfdb69f6a..a20e878d0817d0a75e9c47a44f8765deca99225c 100644 --- a/python/paddle/v2/topology.py +++ b/python/paddle/v2/topology.py @@ -92,7 +92,7 @@ class Topology(object): data_layers = self.data_layers() return [(nm, data_layers[nm].data_type) - for nm in self.proto().input_layer_names if nm in data_layers] + for nm in self.proto().input_layer_names] def get_layer_proto(self, name): for layer in self.__model_config__.layers: