Skip to content

Commit 305d211

Browse files
committed
fix data names
test=develop
1 parent ce1e0d3 commit 305d211

File tree

1 file changed

+2
-1
lines changed
  • python/paddle/fluid/layers

1 file changed

+2
-1
lines changed

python/paddle/fluid/layers/io.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -598,14 +598,15 @@ def __set_paddle_reader__(paddle_reader):
598598
lod_level=lod_level))
599599
counter += 1
600600

601+
data_names = [feed_data.name for feed_data in actual_feed_list]
601602
feeder = DataFeeder(
602603
feed_list=actual_feed_list, place=core.CPUPlace())
603604
paddle_reader = feeder.decorate_reader(
604605
paddle_reader, multi_devices=False)
605606

606607
def __tensor_provider__():
607608
for slots in paddle_reader():
608-
yield [slots[str(idx)] for idx in six.moves.xrange(counter)]
609+
yield [slots[data_name] for data_name in data_names]
609610

610611
__set_tensor_provider__(__tensor_provider__)
611612

0 commit comments

Comments
 (0)