update code

2022-04-22 12:03:17 +08:00 · 2022-04-22 12:03:17 +08:00 · 15997ea9d0
parent f2982e5e47
commit 15997ea9d0
1 changed files with 1 additions and 1 deletions
--- a/ppcls/engine/train/utils.py
+++ b/ppcls/engine/train/utils.py
@ -64,7 +64,7 @@ def log_info(trainer, batch_size, epoch_id, iter_id):

    for i, lr in enumerate(trainer.lr_sch):
        logger.scaler(
-            name="lr_{}".format(i + 1),
+            name="lr({})".format(lr.__class__.__name__),
            value=lr.get_lr(),
            step=trainer.global_step,
            writer=trainer.vdl_writer)