Skip to content

Commit 03cf701

Browse files
authored
use key-value to init dataloader (#11067)
1 parent 9b1c166 commit 03cf701

File tree

1 file changed

+16
-16
lines changed

1 file changed

+16
-16
lines changed

paddlenlp/data/dist_dataloader.py

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -94,22 +94,22 @@ def __init__(
9494

9595
if self._need_data:
9696
self._dataloader = paddle.io.DataLoader(
97-
dataset,
98-
feed_list,
99-
places,
100-
return_list,
101-
batch_sampler,
102-
batch_size,
103-
shuffle,
104-
drop_last,
105-
collate_fn,
106-
num_workers,
107-
use_buffer_reader,
108-
prefetch_factor,
109-
use_shared_memory,
110-
timeout,
111-
worker_init_fn,
112-
persistent_workers,
97+
dataset=dataset,
98+
feed_list=feed_list,
99+
places=places,
100+
return_list=return_list,
101+
batch_sampler=batch_sampler,
102+
batch_size=batch_size,
103+
shuffle=shuffle,
104+
drop_last=drop_last,
105+
collate_fn=collate_fn,
106+
num_workers=num_workers,
107+
use_buffer_reader=use_buffer_reader,
108+
prefetch_factor=prefetch_factor,
109+
use_shared_memory=use_shared_memory,
110+
timeout=timeout,
111+
worker_init_fn=worker_init_fn,
112+
persistent_workers=persistent_workers,
113113
)
114114

115115
self._lazy_dataloader_iter = None

0 commit comments

Comments
 (0)