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 adalah bagian dari kursus
Model AI yang Dapat Diskalakan dengan PyTorch Lightning
Petunjuk latihan
- Buat kelas bernama
ClassifierModelyang menurunkan daripl.LightningModule. - Inisialisasi kelas induk untuk memanfaatkan kapabilitas LightningModule.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
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)