Log lr for param groups (#159)

2023-01-09 04:57:14 +05:30
parent e79ea1666c
commit 9a2f67b3b4
2 changed files with 4 additions and 2 deletions
--- a/ultralytics/yolo/utils/callbacks/wb.py
+++ b/ultralytics/yolo/utils/callbacks/wb.py
@ -25,6 +25,7 @@ def on_fit_epoch_end(trainer):

 def on_train_epoch_end(trainer):
    wandb.run.log(trainer.label_loss_items(trainer.tloss, prefix="train"), step=trainer.epoch + 1)
+    wandb.run.log(trainer.lr, step=trainer.epoch + 1)
    if trainer.epoch == 1:
        wandb.run.log({f.stem: wandb.Image(str(f))
                       for f in trainer.save_dir.glob('train_batch*.jpg')},