can observe lr

Author: Hiroshiba

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"])

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',