ComeçarComece de graça

Ajuste da taxa de aprendizado

Neste exercício, seu objetivo é encontrar a taxa de aprendizado ideal para que o otimizador encontre o mínimo da função não convexa \(x^{4} + x^{3} - 5x^{2}\) em dez etapas.

Você deve testar três valores diferentes de taxa de aprendizado. Nesse problema, teste valores de taxas de aprendizado entre 0,001 e 0,1.

Você recebeu a função optimize_and_plot(), que usa a taxa de aprendizado como primeiro argumento. Essa função executará 10 etapas do otimizador SGD e exibirá os resultados.

Este exercício faz parte do curso

Introdução ao Aprendizado Profundo com o PyTorch

Ver curso

Exercício interativo prático

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

# Try a first learning rate value
lr0 = ____
optimize_and_plot(lr=lr0)
Editar e executar o código