ComenzarEmpieza gratis

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

Ver curso

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)
Editar y ejecutar código