From 287ad43c58b07c5df4c57234acc4b7a5764a8ac4 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 6 Apr 2019 17:06:37 +0200 Subject: [PATCH] updates --- test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test.py b/test.py index 89a3ea06..3abe8303 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=0, + num_workers=4, pin_memory=False, collate_fn=dataset.collate_fn)