updates

2020-01-11 20:15:41 -08:00 · 2020-01-11 20:15:41 -08:00 · 01d485831f
commit 01d485831f
parent 77034467f6
1 changed files with 1 additions and 2 deletions
--- a/train.py
+++ b/train.py
@ -174,8 +174,7 @@ def train():
                                  augment=True,
                                  hyp=hyp,  # augmentation hyperparameters
                                  rect=opt.rect,  # rectangular training
-                                  image_weights=False,
-                                  cache_labels=epochs > 10,
+                                  cache_labels=True,
                                  cache_images=opt.cache_images)

    # Dataloader