Den PyTorch-Optimierer verwenden
Bisher hast du die Gewichte eines Netzes manuell aktualisiert und so einen Einblick bekommen, wie das Training im Hintergrund funktioniert. Allerdings ist diese Methode für tiefe Netze mit vielen Schichten nicht skalierbar.
Zum Glück gibt es in PyTorch den SGD-Optimierer, der diesen Prozess in nur wenigen Codezeilen effizient automatisiert. Jetzt schließt du die Trainingsschleife ab, indem du die Gewichte mit einem PyTorch-Optimizer aktualisierst.
Es wurde ein neuronales Netz erstellt und als model
-Variable bereitgestellt. Dieses Modell wurde verwendet, um einen Vorwärtsdurchlauf auszuführen und den Tensor der Vorhersagen pred
zu erstellen. Der One-Hot-kodierte Tensor heißt target
und die Kreuzentropieverlust-Funktion wird als criterion
gespeichert.
torch.optim
als optim
und torch.nn
als nn
wurden bereits für dich geladen.
Diese Übung ist Teil des Kurses
Einführung in Deep Learning mit PyTorch
Interaktive Übung
Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.
# Create the optimizer
optimizer = ____