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

cleanup train.py

This commit is contained in:
Glenn Jocher 2018-10-09 19:28:27 +02:00
parent 0cc8f2be01
commit e7cd5d01c4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -156,8 +156,8 @@ def main(opt):
s = ('%10s%10s' + '%10.3g' * 14) % ( s = ('%10s%10s' + '%10.3g' * 14) % (
'%g/%g' % (epoch, opt.epochs - 1), '%g/%g' % (i, len(dataloader) - 1), rloss['x'], '%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['y'], rloss['w'], rloss['h'], rloss['conf'], rloss['cls'],
rloss['loss'], mean_precision, mean_recall, model.losses['nT'], TP.sum(), rloss['loss'], mean_precision, mean_recall, model.losses['nT'], rloss['TP'],
FP.sum(), FN.sum(), time.time() - t1) rloss['FP'], rloss['FN'], time.time() - t1)
t1 = time.time() t1 = time.time()
print(s) print(s)