capitalize

ultralytics · Jan 15, 2021 · 5c7c44f · 5c7c44f
1 parent fab241b
commit 5c7c44f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]