ComeçarComece de graça

Otimizadores

É hora de explorar os diferentes otimizadores que você pode usar para treinar seu modelo.

Uma função personalizada chamada train_model(optimizer, net, num_epochs) foi definida para você. Ela usa o otimizador, o modelo e o número de épocas como entradas, executa os loops de treinamento e imprime a perda de treinamento no final.

Vamos usar o train_model() para executar alguns treinamentos curtos com diferentes otimizadores e comparar os resultados!

Este exercício faz parte do curso

Aprendizagem profunda intermediária com PyTorch

Ver curso

Exercício interativo prático

Experimente este exercício completando este código de exemplo.

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,
)
Editar e executar o código