diff --git a/ppcls/engine/train/utils.py b/ppcls/engine/train/utils.py index dc1868bee192e0245feee9fcbfd200900059f160..ca211ff932f19ca63804a5a1ff52def5eb89477f 100644 --- 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)