init metoduna hakim ol
PyTorch Lightning'de __init__ metodu, LightningModule kurulumunun kalbidir. Eğitimden önce model katmanlarını, parametreleri ve başlangıç yapılandırmasını burada tanımlarsın. Bu kurulum adımını net bir şekilde ayırarak, PyTorch Lightning projelerini bakımını kolaylaştırır ve ölçeklenebilir hale getirir. Bu egzersizde, sınıflandırma modelin için sınıfı başlatmaya odaklanacaksın.
Bu egzersiz
PyTorch Lightning ile Ölçeklenebilir AI Modelleri
kursunun bir parçasıdırEgzersiz talimatları
pl.LightningModulesınıfından kalıtım alanClassifierModeladlı bir sınıf oluştur.- LightningModule'un yeteneklerinden yararlanmak için ebeveyn sınıfı başlat.
Uygulamalı interaktif egzersiz
Bu örnek kodu tamamlayarak bu egzersizi bitirin.
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)