ÎncepețiÎncepe gratuit

Metoda init în detaliu

În PyTorch Lightning, metoda __init__ reprezintă nucleul configurării LightningModule-ului tău. Aici definești straturile modelului, parametrii și orice configurație inițială necesară înainte de antrenare. Prin separarea clară a acestui pas de configurare, PyTorch Lightning îți face mai ușoară menținerea și scalarea proiectelor. În acest exercițiu, te vei concentra pe inițializarea clasei pentru modelul tău de clasificare.

Acest exercițiu face parte din cursul

Modele AI scalabile cu PyTorch Lightning

Vezi cursul

Instrucțiuni pentru exercițiu

  • Creează o clasă numită ClassifierModel care moștenește din pl.LightningModule.
  • Inițializează clasa părinte pentru a beneficia de capabilitățile LightningModule.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

import lightning.pytorch as pl
import torch.nn as nn

# Create the class
class ClassifierModel(____):
    # Create init method
    def __init__(self, input_dim, output_dim):
        ____
        self.classifier = nn.Linear(input_dim, output_dim)
Editează și rulează codul