提交 15997ea9 编写于 作者: H HydrogenSulfate

update code

上级 f2982e5e
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册