SequentialLR

On This Page

SequentialLR¶

class modelzoo.common.pytorch.optim.lr_scheduler.SequentialLR (optimizer, *args, **kwargs)

Receives the list of schedulers that is expected to be called sequentially during optimization process and milestone points that provides exact intervals to reflect which scheduler is supposed to be called at a given step. This class is a wrapper around https://pytorch.org/docs/stable/generated/torch.optim.lr_scheduler.SequentialLR.html#torch.optim.lr_scheduler.SequentialLR

Parameters:

optimizer – Wrapped optimizer

schedulers (list) – List of chained schedulers

milestones (list) – List of integers that reflects milestone points

last_epoch (int) – The index of last epoch. Default: -1

`load_state_dict` (state_dict: dict)¶

Loads the schedulers state.

Parameters:: state_dict (dict) – scheduler state. Should be an object returned from a call to state_dict().