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
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
# Create the optimizer
optimizer = ____