Rularea Lightning Trainer
Acum că ai modelul definit, ești pregătit să configurezi Lightning Trainer pentru a rula o buclă de antrenament. Acest exercițiu te ajută să înțelegi cum să integrezi modelul, data loader-ele și configurațiile de antrenament – stăpânirea acestei configurări este esențială înainte de a trece la antrenamente mai avansate.
Bibliotecile torch și lightning.pytorch, importată ca pl, au fost preîncărcate pentru tine. Un model a fost deja inițializat, folosind clasa pe care ai definit-o în exercițiul anterior.
Acest exercițiu face parte din cursul
Modele AI scalabile cu PyTorch Lightning
Instrucțiuni pentru exercițiu
- Configurează
Trainer-ul cu 3 epoci. - Pornește o sesiune de antrenament cu
model-ul predefinit.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
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)