LRScheduler — PyTorch 2.7 documentation (original) (raw)

class torch.optim.lr_scheduler.LRScheduler(optimizer, last_epoch=-1)[source][source]¶

Adjusts the learning rate during optimization.

get_last_lr()[source][source]¶

Return last computed learning rate by current scheduler.

Return type

list[float]

get_lr()[source][source]¶

Compute learning rate using chainable form of the scheduler.

Return type

list[float]

load_state_dict(state_dict)[source][source]¶

Load the scheduler’s state.

Parameters

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

state_dict()[source][source]¶

Return the state of the scheduler as a dict.

It contains an entry for every variable in self.__dict__ which is not the optimizer.

step(epoch=None)[source][source]¶

Perform a step.