Začněte nyníZačněte zdarma

Experimentování s momentem

V tomto cvičení je tvým cílem najít optimální hodnotu momentu tak, aby optimalizátor dokázal ve 20 krocích nalézt minimum následující nekonvexní funkce \(x^{4} + x^{3} - 5x^{2}\). Vyzkoušíš dvě různé hodnoty momentu. Rychlost učení je v tomto případě pevně nastavena na 0,01.

Máš k dispozici funkci optimize_and_plot(), která přijímá jako vstup parametr momentum. Tato funkce spustí 20 kroků SGD optimalizátoru a zobrazí výsledky.

Toto cvičení je součástí kurzu

Introduction to Deep Learning with PyTorch

Zobrazit kurz

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

# Try a first value for momentum
mom0 = ____
optimize_and_plot(momentum=mom0)
Upravit a spustit kód