Оптимизаторы
Пришло время познакомиться с различными оптимизаторами, которые можно использовать для обучения модели.
Для вас уже определена пользовательская функция train_model(optimizer, net, num_epochs). Она принимает оптимизатор, модель и количество эпох, запускает циклы обучения и выводит значение ошибки в конце.
Давайте воспользуемся train_model(), чтобы провести несколько коротких запусков обучения с разными оптимизаторами и сравнить результаты!
Это упражнение является частью курса
Глубокое обучение на 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,
)