Mulai sekarangMulai gratis

Mengenal LightningModule

Bersiaplah membangun LightningModule pertama Anda! Dalam latihan praktis ini, Anda akan menyiapkan struktur inti alur kerja klasifikasi. Anda akan mendefinisikan sebuah layer linear, meneruskan data melaluinya di metode forward, dan menghitung loss pada langkah pelatihan. Struktur yang rapi ini memberi Anda dasar yang kuat untuk mulai bereksperimen dengan model Anda.

torch dan lightning.pytorch yang diimpor sebagai pl telah dimuat sebelumnya untuk Anda.

Latihan ini merupakan bagian dari kursus

Model AI yang Dapat Diskalakan dengan PyTorch Lightning

Lihat Kursus

Instruksi latihan

  • Definisikan kelas LightModel yang mewarisi pl.LightningModule.
  • Definisikan sebuah layer linear untuk mentransformasikan masukan Anda, dengan asumsi fitur masukan berjumlah 16 dan terdapat 10 kelas keluaran.

Latihan interaktif langsung praktik

Cobalah latihan ini dengan melengkapi kode contoh ini.

# Define the model class
class LightModel(____):
  	# Define a linear layer to transform your input
    def __init__(self):
        super().__init__()
        self.layer = ____
    def forward(self, x):
        return self.layer(x)
    def training_step(self, batch, batch_idx):
        x, y = batch
        logits = self(x)
        loss = torch.nn.functional.cross_entropy(logits, y)
        return loss
Edit dan Jalankan Kode