Utilizarea LightningModule
LightningModule se află în centrul abordării PyTorch Lightning pentru structurarea proiectelor de învățare profundă. Acesta centralizează arhitectura modelului, pasajele forward și logica de antrenament, lăsând orchestrarea buclelor de antrenament și configurarea dispozitivelor în seama Trainer-ului.
Care afirmație descrie cel mai bine rolul principal al LightningModule în PyTorch Lightning?
Acest exercițiu face parte din cursul
Modele AI scalabile cu PyTorch Lightning
Exercițiu interactiv practic
Transformă teoria în practică cu unul dintre exercițiile noastre interactive
Începe exercițiul