f7c8d19df1324bd38ed58885e545109ed80110a0,torchbearer/callbacks/torch_scheduler.py,TorchScheduler,on_end_epoch,#TorchScheduler#Any#,32

Before Change

                current = get_metric("Scheduler", state, self._monitor)
                if current is None:
                    return
                self._scheduler.step(current, epoch=state[torchbearer.EPOCH])
            else:
                self._scheduler.step(epoch=state[torchbearer.EPOCH])

After Change

                current = get_metric("Scheduler", state, self._monitor)
                if current is None:
                    return
                self._step(state, current)
            else:
                self._step(state)

Instances