We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ce1e0d3 commit 305d211Copy full SHA for 305d211
python/paddle/fluid/layers/io.py
@@ -598,14 +598,15 @@ def __set_paddle_reader__(paddle_reader):
598
lod_level=lod_level))
599
counter += 1
600
601
+ data_names = [feed_data.name for feed_data in actual_feed_list]
602
feeder = DataFeeder(
603
feed_list=actual_feed_list, place=core.CPUPlace())
604
paddle_reader = feeder.decorate_reader(
605
paddle_reader, multi_devices=False)
606
607
def __tensor_provider__():
608
for slots in paddle_reader():
- yield [slots[str(idx)] for idx in six.moves.xrange(counter)]
609
+ yield [slots[data_name] for data_name in data_names]
610
611
__set_tensor_provider__(__tensor_provider__)
612
0 commit comments