92ec6f9f8790d0176fbafbf711d5a541b5c09e60,catalyst/callbacks/scheduler.py,SchedulerCallback,_scheduler_step,#Any#Any#,114

Before Change

            lr = scheduler.optimizer.param_groups[0]["lr"]
        else:
            scheduler.step()
            lr = scheduler.get_lr()[0]

        momentum = get_optimizer_momentum(scheduler.optimizer)

        return lr, momentum

    def step_batch(self, runner: "IRunner") -> None:
        Update learning rate and momentum in runner.

After Change

        lr_list = [
            param_group["lr"]
            for param_group in scheduler.optimizer.param_groups
        ]
        momentum_list = get_optimizer_momentum_list(scheduler.optimizer)
        return lr_list, momentum_list

Instances