Update get_worker_info API (#29190)

* Update get_worker_info API

* Update dataloader_iter.py

* Update dataloader_iter.py

* Update dataloader_iter.py
revert-31562-mean
mls1999725 5 years ago committed by GitHub
parent c59b4f28a2
commit 0aedd463ee
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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

Loading…
Cancel
Save