Spuštění Lightning Traineru
Teď, když máš model definovaný, je čas nakonfigurovat Lightning Trainer a spustit tréninkovou smyčku. Toto cvičení ti ukáže, jak propojit model, data loadery a nastavení trénování – zvládnutí tohoto kroku je klíčové před přechodem k pokročilejšímu trénování.
torch a lightning.pytorch, importované jako pl, jsou pro tebe předem načtené. model je již inicializovaný, a to pomocí třídy, kterou jsi definoval/a v předchozím cvičení.
Toto cvičení je součástí kurzu
Škálovatelné modely AI s PyTorch Lightning
Pokyny k cvičení
- Nakonfiguruj
Trainerna 3 epochy. - Spusť trénování s předem definovaným
model.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
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)