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