(self, param, param_grad, param_name, cur_loss=None)
| 20 | self.lr_scheduler = SchedulerInitializer(scheduler, lr=lr)() |
| 21 | |
| 22 | def __call__(self, param, param_grad, param_name, cur_loss=None): |
| 23 | return self.update(param, param_grad, param_name, cur_loss) |
| 24 | |
| 25 | def step(self): |
| 26 | """Increment the optimizer step counter by 1""" |