Experimentar con la tasa de aprendizaje
En este ejercicio, tu objetivo es encontrar la tasa de aprendizaje óptima de tal manera que el optimizador pueda encontrar el mínimo de la función no convexa \(x^{4} + x^{3} - 5x^{2}\) en diez pasos.
Experimentarás con tres valores diferentes de tasa de aprendizaje. Para este problema, prueba valores de tasa de aprendizaje entre 0,001 y 0,1.
Se te proporciona la función optimize_and_plot()
, que toma la tasa de aprendizaje como primer argumento. Esta función ejecutará 10 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 learning rate value
lr0 = ____
optimize_and_plot(lr=lr0)