From b33860d8d75674654a160c8da6f517fe07617dff Mon Sep 17 00:00:00 2001 From: Yizhou Wang Date: Sun, 6 Feb 2022 17:55:51 -0800 Subject: [PATCH] add lr to tensorboard --- tools/train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tools/train.py b/tools/train.py index eaacb14..9bf5883 100644 --- a/tools/train.py +++ b/tools/train.py @@ -251,6 +251,7 @@ def parse_args(): writer.add_scalar('loss/loss_ave', loss_ave, iter_count) writer.add_scalar('time/time_load', load_time, iter_count) writer.add_scalar('time/time_back', back_time, iter_count) + writer.add_scalar('param/param_lr', scheduler.get_last_lr()[0], iter_count) if stacked_num is not None: confmap_pred = confmap_preds[stacked_num - 1].cpu().detach().numpy()