From 0aedd463ee6db847885ef4052f2c94ef163e8575 Mon Sep 17 00:00:00 2001 From: mls1999725 <43207078+mls1999725@users.noreply.github.com> Date: Wed, 2 Dec 2020 14:27:14 +0800 Subject: [PATCH] Update get_worker_info API (#29190) * Update get_worker_info API * Update dataloader_iter.py * Update dataloader_iter.py * Update dataloader_iter.py --- .../fluid/dataloader/dataloader_iter.py | 26 +++++++++---------- 1 file changed, 13 insertions(+), 13 deletions(-) diff --git a/python/paddle/fluid/dataloader/dataloader_iter.py b/python/paddle/fluid/dataloader/dataloader_iter.py index ea89b09d2b..31ef3bd7bb 100644 --- a/python/paddle/fluid/dataloader/dataloader_iter.py +++ b/python/paddle/fluid/dataloader/dataloader_iter.py @@ -153,8 +153,8 @@ def get_worker_info(): .. code-block:: python import math + import paddle import numpy as np - import paddle.fluid as fluid from paddle.io import IterableDataset, DataLoader, get_worker_info class SplitedIterableDataset(IterableDataset): @@ -178,18 +178,18 @@ def get_worker_info(): for i in range(iter_start, iter_end): yield np.array([i]) - place = fluid.CPUPlace() - with fluid.dygraph.guard(place): - dataset = SplitedIterableDataset(start=2, end=9) - dataloader = DataLoader( - dataset, - places=place, - num_workers=2, - batch_size=1, - drop_last=True) - - print(list(dataloader)) - # outputs: [2, 5, 3, 6, 4, 7] + place = paddle.CPUPlace() + dataset = SplitedIterableDataset(start=2, end=9) + dataloader = DataLoader( + dataset, + places=place, + num_workers=2, + batch_size=1, + drop_last=True) + + for data in dataloader: + print(data) + # outputs: [2, 5, 3, 6, 4, 7] """ return _worker_info -- GitLab