Optimerare
Nu är det dags att utforska de olika optimerare du kan använda för att träna din modell.
En anpassad funktion, train_model(optimizer, net, num_epochs), har redan definierats åt dig. Den tar emot optimeraren, modellen och antalet epoker som indata, kör träningsslingorna och skriver ut träningsförlusten när den är klar.
Använd train_model() för att köra några kortare träningar med olika optimerare och jämföra resultaten!
Den här övningen är en del av kursen
Fördjupad djupinlärning med PyTorch
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
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,
)