Comece agoraComece grátis

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 exercicio faz parte do curso

Aprendizagem profunda intermediária com PyTorch

Ver curso

exercicio interativo prático

Tente este exercicio 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 Código