diff --git a/dataset/dataset.py b/dataset/dataset.py index 0ea51076a6b3d6382b5c9b6d18902bbb6098e096..f064c404c9126088164e11b1040bdf5a93362d41 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=1, + num_workers=8, 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=1, + num_workers=8, collate_fn=None, pin_memory=False, )