From a34a760d0fb78ca29289ff432667af28fc2054ec Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 5 Apr 2019 16:26:42 +0200
Subject: [PATCH] updates

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

diff --git a/train.py b/train.py
index 83eb7ccc..2d3cf8b2 100644
--- a/train.py
+++ b/train.py
@@ -188,7 +188,7 @@ def train(
         # Update best loss
         test_loss = results[4]
         if test_loss < best_loss:
-            best_loss = results[0]
+            best_loss = test_loss
 
         # Save training results
         save = True and not opt.nosave