docs: def get_param(self):
          """Method to get current optimizer's parameter value
          """
          lr_list = self.lr_scheduler.get_lr()
          if len(lr_list) > 1:
              raise ValueError("Optimizer passed to lr_scheduler should have a single param group, "
                               "but currently there are {} param groups".format(len(lr_list)))
          return lr_list[0]