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.
Toto cvičení je součástí kurzu
Introduction to Deep Learning with PyTorch
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
# Create the optimizer
optimizer = ____