From 2acbe9699a51a0efd8ae0dbfdd775c20a628f362 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 28 Jan 2021 18:57:24 -0800 Subject: [PATCH] W&B epoch logging update (#2073) --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index ddb23e34a619..4ec97ae71e16 100644 --- a/train.py +++ b/train.py @@ -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]