Experimentar con el impulso
En este ejercicio, tu objetivo es encontrar el impulso óptimo de tal manera que el optimizador pueda encontrar el mínimo de la siguiente función no convexa \(x^{4} + x^{3} - 5x^{2}\) en 20 pasos. Experimentarás con dos valores de impulso diferentes. Para este problema, la tasa de aprendizaje se fija en 0,01.
Se te proporciona la función optimize_and_plot()
que acepta como entrada el parámetro momentum
. Esta función ejecutará 20 pasos del optimizador SGD y mostrará los resultados.
Este ejercicio forma parte del curso
Introducción al aprendizaje profundo con PyTorch
Ejercicio interactivo práctico
Prueba este ejercicio completando el código de muestra.
# Try a first value for momentum
mom0 = ____
optimize_and_plot(momentum=mom0)