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!
Toto cvičení je součástí kurzu
Intermediate 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.
import torch.optim as optim
net = Net()
# Define the SGD optimizer
optimizer = optim.____(net.parameters(), lr=0.001)
train_model(
optimizer=optimizer,
net=net,
num_epochs=10,
)