ComenzarEmpieza gratis

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

Ver curso

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