BaşlayınÜcretsiz başlayın

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, kursun bir parçasıdır

PyTorch Lightning ile Ölçeklenebilir AI Modelleri

Kursa Göz Atın

Egzersiz talimatları

  • pl.LightningModule sınıfından kalıtım alan ClassifierModel adlı bir sınıf oluştur.
  • LightningModule'un yeteneklerinden yararlanmak için ebeveyn sınıfı başlat.

Uygulamalı etkileşimli egzersiz

Bu egzersizi bu örnek kodu tamamlayarak deneyin.

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)
Kodu Düzenle ve Çalıştır