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
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,
)