can observe lr

Hiroshiba · Nov 29, 2020 · a395291 · a395291
1 parent 7ccc860
commit a395291
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pytorch_trainer/training/extensions/value_observation.py b/pytorch_trainer/training/extensions/value_observation.py
@@ -41,4 +41,4 @@ def observe_lr(optimizer_name='main', observation_key='lr'):
     """
     return observe_value(
         observation_key,
-        lambda trainer: trainer.updater.get_optimizer(optimizer_name).lr)
+        lambda trainer: trainer.updater.get_optimizer(optimizer_name).param_groups[0]["lr"])
diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name='pytorch-trainer',
-    version='1.3.0',
+    version='1.3.1',
     packages=find_packages(),
     url='https://github.com/Hiroshiba/pytorch-trainer',
     author='Kazuyuki Hiroshiba',