Pytorch Изменение скорости обучения в зависимости от количества эпох

Когда я устанавливаю скорость обучения и обнаруживаю, что точность не может увеличиться после тренировки нескольких эпох

optimizer = optim.Adam(model.parameters(), lr = 1e-4)

n_epochs = 10
for i in range(n_epochs):

    // some training here

Если я хочу использовать пошаговое затухание: уменьшать скорость обучения в 10 раз каждые 5 эпох, как я могу это сделать?


pytorch optimization learning-rate
person Shin Yu Wu    schedule 04.02.2020    source источник

Ответы (1)


arrow_upward
25
arrow_downward

Вы можете использовать lr shedular torch.optim.lr_scheduler.StepLR

import torch.optim.lr_scheduler.StepLR
scheduler = StepLR(optimizer, step_size=5, gamma=0.1)

Снижает скорость обучения каждой группы параметров на gamma каждые step_size эпохи см. Документацию здесь Пример из документов

# Assuming optimizer uses lr = 0.05 for all groups
# lr = 0.05     if epoch < 30
# lr = 0.005    if 30 <= epoch < 60
# lr = 0.0005   if 60 <= epoch < 90
# ...
scheduler = StepLR(optimizer, step_size=30, gamma=0.1)
for epoch in range(100):
    train(...)
    validate(...)
    scheduler.step()

Пример:

import torch
import torch.optim as optim

optimizer = optim.SGD([torch.rand((2,2), requires_grad=True)], lr=0.1)
scheduler = optim.lr_scheduler.StepLR(optimizer, step_size=5, gamma=0.1)

for epoch in range(1, 21):
    scheduler.step()
    print('Epoch-{0} lr: {1}'.format(epoch, optimizer.param_groups[0]['lr']))
    if epoch % 5 == 0:print()

Epoch-1 lr: 0.1
Epoch-2 lr: 0.1
Epoch-3 lr: 0.1
Epoch-4 lr: 0.1
Epoch-5 lr: 0.1

Epoch-6 lr: 0.010000000000000002
Epoch-7 lr: 0.010000000000000002
Epoch-8 lr: 0.010000000000000002
Epoch-9 lr: 0.010000000000000002
Epoch-10 lr: 0.010000000000000002

Epoch-11 lr: 0.0010000000000000002
Epoch-12 lr: 0.0010000000000000002
Epoch-13 lr: 0.0010000000000000002
Epoch-14 lr: 0.0010000000000000002
Epoch-15 lr: 0.0010000000000000002

Epoch-16 lr: 0.00010000000000000003
Epoch-17 lr: 0.00010000000000000003
Epoch-18 lr: 0.00010000000000000003
Epoch-19 lr: 0.00010000000000000003
Epoch-20 lr: 0.00010000000000000003

Подробнее How to adjust Learning Rate - torch.optim.lr_scheduler предоставляет несколько методов настройки скорость обучения в зависимости от количества эпох.

person Dishin H Goyani    schedule 04.02.2020
comment
optimizer.step () также необходим, поскольку scheduler.step () контролирует только скорость обучения. писать это на случай, если кто-то это пропустил. - person Stallion; 25.10.2020