From d748bedb1dbce278c14dbd3e968459bdc654ec19 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 9 Oct 2018 19:32:42 +0200 Subject: [PATCH] clean up train.py --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index bc474f5b..9e9d60df 100644 --- a/train.py +++ b/train.py @@ -156,8 +156,8 @@ def main(opt): s = ('%10s%10s' + '%10.3g' * 14) % ( '%g/%g' % (epoch, opt.epochs - 1), '%g/%g' % (i, len(dataloader) - 1), rloss['x'], rloss['y'], rloss['w'], rloss['h'], rloss['conf'], rloss['cls'], - rloss['loss'], mean_precision, mean_recall, model.losses['nT'], rloss['TP'], - rloss['FP'], rloss['FN'], time.time() - t1) + rloss['loss'], mean_precision, mean_recall, model.losses['nT'], model.losses['TP'], + model.losses['FP'], model.losses['FN'], time.time() - t1) t1 = time.time() print(s)