mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

W&B epoch logging update (#2073)

Browse Source
This commit is contained in:
Glenn Jocher 2021-01-28 18:57:24 -08:00 committed by GitHub
parent 630ec06af6
commit 2acbe9699a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -364,7 +364,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
if tb_writer: if tb_writer:
tb_writer.add_scalar(tag, x, epoch) # tensorboard tb_writer.add_scalar(tag, x, epoch) # tensorboard
if wandb: if wandb:
wandb.log({tag: x}) # W&B wandb.log({tag: x}, step=epoch, commit=tag == tags[-1]) # W&B
# Update best mAP # Update best mAP
fi = fitness(np.array(results).reshape(1, -1)) # weighted combination of [P, R, mAP@.5, mAP@.5-.95] fi = fitness(np.array(results).reshape(1, -1)) # weighted combination of [P, R, mAP@.5, mAP@.5-.95]