ComeçarComece de graça

Ajuste do momentum

Neste exercício, seu objetivo é encontrar o momentum ideal para que o otimizador consiga encontrar o mínimo da seguinte função não convexa \(x^{4} + x^{3} - 5x^{2}\) em 20 etapas. Você deve fazer testes com dois valores diferentes de momentum. Nesse problema, a taxa de aprendizado é fixada em 0,01.

Você tem à disposição a função optimize_and_plot(), que aceita como entrada o parâmetro momentum. Essa função executará 20 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 value for momentum
mom0 = ____
optimize_and_plot(momentum=mom0)
Editar e executar o código