Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-08-29 15:49:51 +02:00
parent 408baf66e2
commit 8eb381dc88
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
train.py
View File

@ -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 ------------------------------------------------------------------------------------------------