diff --git a/dataset/dataset.py b/dataset/dataset.py index e06b3f1febd7768d3a447cedd93c99aae1b22734..0ea51076a6b3d6382b5c9b6d18902bbb6098e096 100644 --- a/dataset/dataset.py +++ b/dataset/dataset.py @@ -252,7 +252,7 @@ def load_data_duo(base_dir, batch_size, args, shuffle=True): dataset=train_dataset, batch_size=batch_size, shuffle=shuffle, - num_workers=0, + num_workers=1, collate_fn=None, pin_memory=False, ) @@ -260,7 +260,7 @@ def load_data_duo(base_dir, batch_size, args, shuffle=True): dataset=val_dataset, batch_size=batch_size, shuffle=shuffle, - num_workers=0, + num_workers=1, collate_fn=None, pin_memory=False, )