From 01d485831fa77a31a6425e22bd5e7dd3e78ff5f7 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 11 Jan 2020 20:15:41 -0800
Subject: [PATCH] updates

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

diff --git a/train.py b/train.py
index c36b0e00..373b709c 100644
--- 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