Learning rate schedulers

Contains decorators for torch optimisers to apply LR schedulers as part of the optimisation step.

class vanguard.optimise.schedule.ApplyLearningRateScheduler(scheduler_class, *args, **kwargs)

Apply a torch learning rate scheduler to a torch optimiser.

The scheduler is stepped at each step of optimiser.

Parameters:

• scheduler_class (type[LRScheduler])

• args (Any)

• kwargs (Any)

__init__(scheduler_class, *args, **kwargs)

Parameters:

• scheduler_class (type[LRScheduler]) – The (uninstantiated) torch learning rate scheduler to be used.

• args (Any)

• kwargs (Any)