PolynomialLR

class torch.optim.lr_scheduler.PolynomialLR(optimizer, total_iters=5, power=1.0, last_epoch=-1, verbose=False) [source]

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

Параметры

• optimizer (Optimizer) – Обёртка оптимизатора.
• total_iters (int) – Количество шагов, на которых планировщик уменьшает скорость обучения. По умолчанию: 5.
• power (int) – Степень полинома. По умолчанию: 1.0.
• verbose (bool) – Если True, выводит сообщение в stdout для каждого обновления. По умолчанию: False.

Пример

>>> # Assuming optimizer uses lr = 0.001 for all groups
>>> # lr = 0.001     if epoch == 0
>>> # lr = 0.00075   if epoch == 1
>>> # lr = 0.00050   if epoch == 2
>>> # lr = 0.00025   if epoch == 3
>>> # lr = 0.0       if epoch >= 4
>>> scheduler = PolynomialLR(self.opt, total_iters=4, power=1.0)
>>> for epoch in range(100):
>>>     train(...)
>>>     validate(...)
>>>     scheduler.step()

get_last_lr()

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

load_state_dict(state_dict)

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

Параметры

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

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

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

state_dict()

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

Он содержит запись для каждой переменной в self.__dict__, которая не является оптимизатором.