From e7cd5d01c409132d0e5ae1393ff1eca68ff244c9 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 9 Oct 2018 19:28:27 +0200 Subject: [PATCH] cleanup train.py --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index a605e186..bc474f5b 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'], TP.sum(), - FP.sum(), FN.sum(), time.time() - t1) + rloss['loss'], mean_precision, mean_recall, model.losses['nT'], rloss['TP'], + rloss['FP'], rloss['FN'], time.time() - t1) t1 = time.time() print(s)