core.utils.torchutils.perform_scheduler_step

core.utils.torchutils.perform_scheduler_step(scheduler, loss=None)

Wrapper for scheduler.step(). If scheduler is ReduceLROnPlateau, then scheduler.step(loss) is called, if not, scheduler.step().

Parameters

Name	Type	Description	Default
scheduler		scheduler for learning rate	required
loss		validation loss	None