Uruchamianie Lightning Trainera
Masz już zdefiniowany model – czas skonfigurować Lightning Trainer i uruchomić pętlę treningową. To ćwiczenie pomoże ci zrozumieć, jak połączyć model, loadery danych i ustawienia treningu. Opanowanie tego kroku jest niezbędne przed przejściem do bardziej zaawansowanych zagadnień.
Biblioteki torch i lightning.pytorch (zaimportowana jako pl) są już wczytane. Zmienna model została zainicjalizowana przy użyciu klasy zdefiniowanej w poprzednim ćwiczeniu.
To ćwiczenie jest częścią kursu
Skalowalne modele AI z PyTorch Lightning
Instrukcje do ćwiczenia
- Skonfiguruj
Trainerna 3 epoki. - Uruchom trening z użyciem gotowego obiektu
model.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
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)