From bb3682024efb6ecde7de937d427419b989763b22 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 15 Jun 2019 01:35:55 +0200
Subject: [PATCH] updates

---
 train.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index 70a9354ed2..54fbb8b5bd 100644
--- a/train.py
+++ b/train.py
@@ -140,11 +140,12 @@ def train(
     # plt.savefig('LR.png', dpi=300)
 
     # Dataset
+    rectangular_training = False
     dataset = LoadImagesAndLabels(train_path,
                                   img_size,
                                   batch_size,
                                   augment=True,
-                                  rect=False)
+                                  rect=rectangular_training)
 
     # Initialize distributed training
     if torch.cuda.device_count() > 1:
@@ -156,7 +157,7 @@ def train(
     dataloader = DataLoader(dataset,
                             batch_size=batch_size,
                             num_workers=opt.num_workers,
-                            shuffle=True,  # disable rectangular training if True
+                            shuffle=not rectangular_training,  # Shuffle=True unless rectangular training is used
                             pin_memory=True,
                             collate_fn=dataset.collate_fn)