From 8ae06ad7c346fd85c7414e917a824596703bc422 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 25 Dec 2019 19:58:20 -0800 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index ee9caad2..f6411e27 100644 --- a/train.py +++ b/train.py @@ -208,7 +208,7 @@ def train(): # Test Dataloader if not opt.prebias: - testloader = torch.utils.data.DataLoader(LoadImagesAndLabels(test_path, opt.img_size, batch_size, + testloader = torch.utils.data.DataLoader(LoadImagesAndLabels(test_path, opt.img_size, batch_size * 2, hyp=hyp, rect=True, cache_labels=True,