From 31d807e58984d1bc9361777f15194998c52f3a17 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Thu, 29 Aug 2019 14:20:54 +0200
Subject: [PATCH] weight_decay fix

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

diff --git a/train.py b/train.py
index c005480a..938e19e5 100644
--- a/train.py
+++ b/train.py
@@ -259,7 +259,7 @@ def train():
             # Compute loss
             loss, loss_items = compute_loss(pred, targets, model)
             if torch.isnan(loss):
-                print('WARNING: nan loss detected, ending training')
+                print('WARNING: nan loss detected, ending training', loss_items)
                 return results
 
             # Scale loss by nominal batch_size of 64