Aan de slagGa gratis aan de slag

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

Cursus bekijken

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,
)
Code bewerken en uitvoeren