From 1e62ee2152203b4175a982995389dbb1a74bb05a Mon Sep 17 00:00:00 2001
From: glenn-jocher <glenn.jocher@ultralytics.com>
Date: Wed, 3 Jul 2019 16:17:46 +0200
Subject: [PATCH] updates

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

diff --git a/train.py b/train.py
index b5dbd857..40e2381c 100644
--- a/train.py
+++ b/train.py
@@ -128,7 +128,7 @@ def train(
     # plt.savefig('LR.png', dpi=300)
 
     # Dataset
-    rectangular_training = False
+    rectangular_training = True
     dataset = LoadImagesAndLabels(train_path,
                                   img_size,
                                   batch_size,
@@ -196,7 +196,7 @@ def train(
             imgs = imgs.to(device)
             targets = targets.to(device)
 
-            # Multi-Scale training
+            # Multi-Scale training TODO: short-side to 32-multiple https://github.com/ultralytics/yolov3/issues/358
             if multi_scale:
                 if (i + nb * epoch) / accumulate % 10 == 0:  #  adjust (67% - 150%) every 10 batches
                     img_size = random.choice(range(img_size_min, img_size_max + 1)) * 32