From 658f2a45769c1abda5544abfc00b94d898185301 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 2 Apr 2019 16:15:25 +0200 Subject: [PATCH] updates --- test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test.py b/test.py index 7acaaf5c..f129e813 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)