updates

2019-12-07 22:55:26 -08:00 · 2019-12-07 22:55:26 -08:00 · b81beb0f5f
parent 1f943e886f
commit b81beb0f5f
1 changed files with 1 additions and 0 deletions
--- a/train.py
+++ b/train.py
@ -210,6 +210,7 @@ def train():
    # Test Dataloader
    if not opt.prebias:
        testloader = torch.utils.data.DataLoader(LoadImagesAndLabels(test_path, img_size, batch_size, hyp=hyp,
+                                                                     rect=True,
                                                                     cache_labels=True,
                                                                     cache_images=opt.cache_images),
                                                 batch_size=batch_size,