Skip to content
Snippets Groups Projects
Commit db1764b7 authored by Thomas Hossler's avatar Thomas Hossler Committed by Kai Chen
Browse files

shuffle passed as function argument instead of kwargs (#1693)

parent 625efeb8
No related branches found
No related tags found
No related merge requests found
...@@ -19,8 +19,8 @@ def build_dataloader(dataset, ...@@ -19,8 +19,8 @@ def build_dataloader(dataset,
workers_per_gpu, workers_per_gpu,
num_gpus=1, num_gpus=1,
dist=True, dist=True,
shuffle=True,
**kwargs): **kwargs):
shuffle = kwargs.get('shuffle', True)
if dist: if dist:
rank, world_size = get_dist_info() rank, world_size = get_dist_info()
if shuffle: if shuffle:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment