diff --git a/train.py b/train.py index 2518f13dcbfa..af96a10b3586 100644 --- a/train.py +++ b/train.py @@ -361,7 +361,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, 'epoch': epoch}) # W&B + wandb.log({tag: x, 'Epoch': epoch}) # W&B # Update best mAP fi = fitness(np.array(results).reshape(1, -1)) # weighted combination of [P, R, mAP@.5, mAP@.5-.95]