1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Intermediate Deep Learning with PyTorch
Connected

cvičení

Optimizéry

Je čas prozkoumat různé optimizéry, které můžeš použít při trénování modelu.

Máš k dispozici vlastní funkci train_model(optimizer, net, num_epochs). Přijímá optimizér, model a počet epoch jako vstupní parametry, spustí trénovací smyčky a na konci vypíše trénovací ztrátu.

Použijeme train_model() k několika krátkým tréninkům s různými optimizéry a porovnáme výsledky!

Instrukce 1/3

undefined XP
  • 1
    • Definuj optimizer jako Stochastic Gradient Descent.
  • 2
    • Definuj optimizér jako Root Mean Square Propagation (RMSprop) a předej mu parametry modelu jako první argument.
  • 3
    • Definuj optimizér jako Adaptive Moments Estimation (Adam) a nastav rychlost učení na 0.001.