From f2ed06e9f7bd70197c5c0510f132792103306140 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E5=BC=80=E8=89=AF?= <972398472@qq.com> Date: Sun, 3 Jan 2021 12:04:03 +0800 Subject: [PATCH] print learning rate of param_groups[0] --- .../tiny_detector_demo/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/code/chapter03_object_detection_introduction/tiny_detector_demo/utils.py b/code/chapter03_object_detection_introduction/tiny_detector_demo/utils.py index c536244..9e3f22e 100644 --- a/code/chapter03_object_detection_introduction/tiny_detector_demo/utils.py +++ b/code/chapter03_object_detection_introduction/tiny_detector_demo/utils.py @@ -632,7 +632,7 @@ def adjust_learning_rate(optimizer, scale): """ for param_group in optimizer.param_groups: param_group['lr'] = param_group['lr'] * scale - print("DECAYING learning rate.\n The new LR is %f\n" % (optimizer.param_groups[1]['lr'],)) + print("DECAYING learning rate.\n The new LR is %f\n" % (optimizer.param_groups[0]['lr'],)) def accuracy(scores, targets, k):