class_docstr: Plugin for LinearSchedulerWithWarmup.
class_docstr_tok: ["Plugin","for","LinearSchedulerWithWarmup","."]
code: def store_learning_rate(self):
      optimizer = self.trainer.optimizer
  
      self.current_learning_rate = [
          group["lr"] for group in optimizer.param_groups
      ]
  
      self.current_momentum = [
          group["betas"][0]
          if "betas" in group
          else group.get("momentum", 0)
          for group in optimizer.param_groups
      ]
  
file_path: flair/flair/trainers/plugins/functional/linear_scheduler.py
imports: import logging
  from typing import Any, Dict
  from flair.optim import LinearSchedulerWithWarmup
  from flair.trainers.plugins.base import TrainerPlugin
label_desc: Use image node_id 2 for calling the LinearSchedulerPlugin obj's underlying member method code with example usage: obj.store_learning_rate() without return types