提交 305d211a 编写于 作者: Q Qiao Longfei

fix data names

test=develop
上级 ce1e0d35
...@@ -598,6 +598,7 @@ def _py_reader(capacity, ...@@ -598,6 +598,7 @@ def _py_reader(capacity,
lod_level=lod_level)) lod_level=lod_level))
counter += 1 counter += 1
data_names = [feed_data.name for feed_data in actual_feed_list]
feeder = DataFeeder( feeder = DataFeeder(
feed_list=actual_feed_list, place=core.CPUPlace()) feed_list=actual_feed_list, place=core.CPUPlace())
paddle_reader = feeder.decorate_reader( paddle_reader = feeder.decorate_reader(
...@@ -605,7 +606,7 @@ def _py_reader(capacity, ...@@ -605,7 +606,7 @@ def _py_reader(capacity,
def __tensor_provider__(): def __tensor_provider__():
for slots in paddle_reader(): for slots in paddle_reader():
yield [slots[str(idx)] for idx in six.moves.xrange(counter)] yield [slots[data_name] for data_name in data_names]
__set_tensor_provider__(__tensor_provider__) __set_tensor_provider__(__tensor_provider__)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册