diff --git a/src/suinput/dataset.py b/src/suinput/dataset.py index a581082..0ed0886 100644 --- a/src/suinput/dataset.py +++ b/src/suinput/dataset.py @@ -595,7 +595,7 @@ if __name__ == "__main__": dataloader = DataLoader( dataset, batch_size=256, - num_workers=12, + num_workers=15, worker_init_fn=worker_init_fn, # pin_memory=True, collate_fn=custom_collate,