ПочатиПочніть безкоштовно

Оптимізатори

Час дослідити різні оптимізатори, які ви можете використовувати для тренування моделі.

Для вас визначено користувацьку функцію train_model(optimizer, net, num_epochs). Вона приймає оптимізатор, модель і кількість епох як вхідні параметри, запускає цикли тренування та в кінці виводить тренувальну втрату.

Скористаймося train_model(), щоб запустити кілька коротких тренувань з різними оптимізаторами та порівняти результати!

Ця вправа є частиною курсу

Середній рівень Deep Learning з PyTorch

Переглянути курс

Інтерактивна практична вправа

Спробуйте виконати цю вправу, доповнивши цей зразок коду.

import torch.optim as optim

net = Net()

# Define the SGD optimizer
optimizer = optim.____(net.parameters(), lr=0.001)

train_model(
    optimizer=optimizer,
    net=net,
    num_epochs=10,
)
Редагувати та запускати код