updates

2019-07-03 16:18:08 +02:00 · 2019-07-03 16:18:08 +02:00 · 109991198c
parent 1e62ee2152
commit 109991198c
1 changed files with 1 additions and 1 deletions
--- 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,