W&B epoch logging update (#2073)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-01-28 18:57:24 -08:00
1 parent 630ec06af6
commit 2acbe9699a
1 file changed
+1 -1
+1 -1
View File
@@ -364,7 +364,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
if tb_writer:
tb_writer.add_scalar(tag, x, epoch) # tensorboard
if wandb:
wandb.log({tag: x}) # W&B
wandb.log({tag: x}, step=epoch, commit=tag == tags[-1]) # W&B
# Update best mAP
fi = fitness(np.array(results).reshape(1, -1)) # weighted combination of [P, R, mAP@.5, mAP@.5-.95]