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

cvičení

Použití optimalizátoru PyTorch

Dříve jsi ručně aktualizoval/a váhy sítě a získal/a tak přehled o tom, jak trénování funguje v zákulisí. Tento přístup ale není příliš škálovatelný pro hluboké sítě s mnoha vrstvami.

Naštěstí PyTorch nabízí optimalizátor SGD, který celý tento proces efektivně automatizuje pomocí několika řádků kódu. Nyní dokončíš tréninkovou smyčku aktualizací vah pomocí optimalizátoru PyTorch.

Neuronová síť byla vytvořena a je dostupná jako proměnná model. Tento model byl použit ke spuštění dopředného průchodu a vytvoření tenzoru predikcí pred. One-hot enkódovaný tenzor se jmenuje target a ztrátová funkce cross entropy je uložena jako criterion.

torch.optim jako optim a torch.nn jako nn jsou již načteny.

Pokyny 1/2

undefined XP
    1
    2
  • Pomocí optim vytvoř optimalizátor SGD s learning rate dle vlastního výběru (musí být menší než jedna) pro zadaný model.