Optimizers
Tijd om de verschillende optimizers te verkennen die je kunt gebruiken om je model te trainen.
Er is een aangepaste functie voor je gedefinieerd: train_model(optimizer, net, num_epochs). Deze neemt de optimizer, het model en het aantal epochs als invoer, voert de trainingslussen uit en print aan het eind het trainingverlies.
Laten we train_model() gebruiken om een paar korte trainingen met verschillende optimizers te draaien en de resultaten te vergelijken!
Deze oefening maakt deel uit van de cursus
Gevorderde Deep Learning met PyTorch
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
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,
)