MultiplicativeLR

class torch.optim.lr_scheduler.MultiplicativeLR(optimizer, lr_lambda, last_epoch=-1, verbose=False) [source]

Умножает скорость обучения каждого параметра группы на множитель, указанный в заданной функции. Когда last_epoch = -1, устанавливает начальную скорость обучения как lr.

Параметры

• optimizer (Optimizer) – Оборачиваемый оптимизатор.
• lr_lambda (функция или список) – Функция, вычисляющая множитель, заданный целым параметром эпохи, или список таких функций, по одной для каждой группы в optimizer.param_groups.
• last_epoch (целое число) – Индекс последней эпохи. По умолчанию: -1.
• verbose (булево) – Если True, выводит сообщение в стандартный вывод для каждого обновления. По умолчанию: False.

Пример

>>> lmbda = lambda epoch: 0.95
>>> scheduler = MultiplicativeLR(optimizer, lr_lambda=lmbda)
>>> for epoch in range(100):
>>>     train(...)
>>>     validate(...)
>>>     scheduler.step()

get_last_lr()

Возвращает последнюю вычисленную скорость обучения текущим планировщиком.

load_state_dict(state_dict) [source]

Загружает состояние планировщика.

Параметры

state_dict (словарь) – состояние планировщика. Должно быть объектом, возвращенным при вызове state_dict().

print_lr(is_verbose, group, lr, epoch=None)

Отображает текущую скорость обучения.

state_dict() [source]

Возвращает состояние планировщика в виде dict.

Он содержит запись для каждой переменной в self.__dict__, которая не является оптимизатором. Функции лямбда-скорости обучения будут сохранены только в том случае, если они являются вызываемыми объектами, а не функциями или лямбдами.