De PyTorch-optimizer gebruiken
Eerder heb je het gewicht van een netwerk handmatig bijgewerkt, zodat je inzicht kreeg in hoe training achter de schermen werkt. Deze aanpak schaalt echter niet voor diepe netwerken met veel lagen.
Gelukkig biedt PyTorch de SGD-optimizer, die dit proces efficiënt automatiseert in slechts een paar regels code. Nu ga je de training loop afronden door de gewichten bij te werken met een PyTorch-optimizer.
Er is een neuraal netwerk gemaakt en beschikbaar als de variabele model. Dit model is gebruikt om een forward pass uit te voeren en de tensor met voorspellingen pred te maken. De one-hot-gecodeerde tensor heet target en de cross-entropy-verliesfunctie is opgeslagen als criterion.
torch.optim als optim, en torch.nn als nn zijn al voor je geladen.
Deze oefening maakt deel uit van de cursus
Introductie tot Deep Learning met PyTorch
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
# Create the optimizer
optimizer = ____