Den Lightning Trainer starten
Jetzt, wo du dein Modell definiert hast, kannst du den Lightning- Trainer
so einrichten, dass ein Trainingslauf durchläuft. Diese Übung hilft dir dabei, zu verstehen, wie du dein Modell, deine Datenlader und deine Trainingskonfigurationen integrieren kannst. Bevor du mit fortgeschritteneren Trainingsschritten weitermachst, solltest du diese Einrichtung unbedingt beherrschen.
Die Dateien „ torch
“ und „ lightning.pytorch
“, importiert als „ pl
“, wurden bereits für dich vorinstalliert. Ein „ model
” wurde schon mit der Klasse, die du in der letzten Übung gemacht hast, gestartet.
Diese Übung ist Teil des Kurses
Skalierbare KI-Modelle mit PyTorch Lightning
Anleitung zur Übung
- Konfigurier das „
Trainer
“ mit 3 Epochen. - Starte einen Trainingslauf mit dem vordefinierten „
model
“.
Interaktive Übung
Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.
from lightning.pytorch import Trainer
# Configure the Trainer with max epochs
trainer = ____(max_epochs=____, devices=1)
# Start training with the predefined model
trainer.____(____, train_dataloader, val_dataloader)