diff --git a/train.py b/train.py
index 40e2381c..ea00fc68 100644
--- a/train.py
+++ b/train.py
@@ -128,7 +128,7 @@ def train(
     # plt.savefig('LR.png', dpi=300)
 
     # Dataset
-    rectangular_training = True
+    rectangular_training = False
     dataset = LoadImagesAndLabels(train_path,
                                   img_size,
                                   batch_size,