From b81beb0f5f0cf74c2ec8065ce9bde78c2c4e1841 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 7 Dec 2019 22:55:26 -0800
Subject: [PATCH] updates

---
 train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/train.py b/train.py
index ef45c5dd..ff02e154 100644
--- 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,