updates

2019-12-14 15:40:01 -08:00 · 2019-12-14 15:40:01 -08:00 · 8c13717f48
parent eb70e4b751
commit 8c13717f48
1 changed files with 2 additions and 1 deletions
--- 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),