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