From 8eb381dc8857b2fc892a94565db6a6ff277a2200 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 29 Aug 2019 15:49:51 +0200 Subject: [PATCH] updates --- train.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/train.py b/train.py index 5df73657..e85d9343 100644 --- a/train.py +++ b/train.py @@ -283,9 +283,6 @@ def train(): s = ('%10s' * 2 + '%10.3g' * 6) % ( '%g/%g' % (epoch, epochs - 1), '%.3gG' % mem, *mloss, len(targets), img_size) pbar.set_description(s) - if torch.isnan(loss): - print('WARNING: nan loss detected, ending training', loss_items) - return results # end batch ------------------------------------------------------------------------------------------------