LoslegenKostenlos loslegen

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

Kurs anzeigen

Interaktive Übung

Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.

# Create the optimizer
optimizer = ____
Code bearbeiten und ausführen