glenn-jocher
commited on
Commit
•
5de4e25
1
Parent(s):
224e9a1
update tensorboard metric
Browse files
train.py
CHANGED
@@ -308,7 +308,7 @@ def train(hyp):
|
|
308 |
# Tensorboard
|
309 |
if tb_writer:
|
310 |
tags = ['train/giou_loss', 'train/obj_loss', 'train/cls_loss',
|
311 |
-
'metrics/precision', 'metrics/recall', 'metrics/mAP_0.5', 'metrics/
|
312 |
'val/giou_loss', 'val/obj_loss', 'val/cls_loss']
|
313 |
for x, tag in zip(list(mloss[:-1]) + list(results), tags):
|
314 |
tb_writer.add_scalar(tag, x, epoch)
|
|
|
308 |
# Tensorboard
|
309 |
if tb_writer:
|
310 |
tags = ['train/giou_loss', 'train/obj_loss', 'train/cls_loss',
|
311 |
+
'metrics/precision', 'metrics/recall', 'metrics/mAP_0.5', 'metrics/mAP_0.5:0.95',
|
312 |
'val/giou_loss', 'val/obj_loss', 'val/cls_loss']
|
313 |
for x, tag in zip(list(mloss[:-1]) + list(results), tags):
|
314 |
tb_writer.add_scalar(tag, x, epoch)
|