Mulai sekarangMulai gratis

Menguasai metode init

Dalam PyTorch Lightning, metode __init__ adalah inti dari penyiapan LightningModule Anda. Di sinilah Anda mendefinisikan lapisan model, parameter, dan konfigurasi awal apa pun sebelum pelatihan. Dengan memisahkan langkah penyiapan ini secara jelas, PyTorch Lightning memudahkan pemeliharaan dan penskalaan proyek Anda. Pada latihan ini, Anda akan berfokus pada inisialisasi kelas untuk model klasifikasi Anda.

Latihan ini merupakan bagian dari kursus

Model AI yang Dapat Diskalakan dengan PyTorch Lightning

Lihat Kursus

Instruksi latihan

  • Buat kelas bernama ClassifierModel yang menurunkan dari pl.LightningModule.
  • Inisialisasi kelas induk untuk memanfaatkan kapabilitas LightningModule.

Latihan interaktif langsung praktik

Cobalah latihan ini dengan melengkapi kode contoh ini.

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)
Edit dan Jalankan Kode