updates

2019-04-05 16:26:42 +02:00 · 2019-04-05 16:26:42 +02:00 · a34a760d0f
commit a34a760d0f
parent 54e43b9ad6
1 changed files with 1 additions and 1 deletions
--- 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