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

Mastering the init method

In PyTorch Lightning, the __init__ method is the heart of your LightningModule setup. It's where you define model layers, parameters, and any initial configuration before training. By cleanly separating this setup step, PyTorch Lightning makes it easier to maintain and scale your projects. In this exercise, you'll focus on initializing the class for your classification model.

Bu egzersiz

Scalable AI Models with PyTorch Lightning

kursunun bir parçasıdır
Kursu Görüntüle

Egzersiz talimatları

  • Create a class named ClassifierModel that inherits from pl.LightningModule.
  • Initialize the parent class to leverage LightningModule's capabilities.

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