diff --git a/test.py b/test.py index 455ae0de..136dc112 100644 --- a/test.py +++ b/test.py @@ -47,7 +47,7 @@ def test( dataset = LoadImagesAndLabels(test_path, img_size=img_size) dataloader = DataLoader(dataset, batch_size=batch_size, - num_workers=4, + num_workers=0, pin_memory=False, collate_fn=dataset.collate_fn)