МультипликативноеLR

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

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

Параметры:

• 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__, которая не является оптимизатором. Функции лямбда скорости обучения будут сохранены только в том случае, если они являются вызываемыми объектами, а не функциями или лямбдами.