From 8c13717f4855299f18456f872b6d1b503719fe1b Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 14 Dec 2019 15:40:01 -0800 Subject: [PATCH] updates --- train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/train.py b/train.py index 4fece807..4c1eaeeb 100644 --- a/train.py +++ b/train.py @@ -208,7 +208,8 @@ def train(): # Test Dataloader if not opt.prebias: - testloader = torch.utils.data.DataLoader(LoadImagesAndLabels(test_path, img_size, batch_size, hyp=hyp, + testloader = torch.utils.data.DataLoader(LoadImagesAndLabels(test_path, opt.img_size, batch_size, + hyp=hyp, rect=True, cache_labels=True, cache_images=opt.cache_images),