НачатьНачать бесплатно

Метод __init__: основы инициализации

В PyTorch Lightning метод __init__ — это основа настройки вашего LightningModule. Именно здесь определяются слои модели, параметры и начальная конфигурация — всё это происходит до начала обучения. Благодаря чёткому разделению этапа настройки PyTorch Lightning упрощает поддержку и масштабирование проектов. В этом упражнении вы сосредоточитесь на инициализации класса для модели классификации.

Это упражнение является частью курса

Масштабируемые модели ИИ с PyTorch Lightning

Посмотреть курс

Инструкции к упражнению

  • Создайте класс с именем ClassifierModel, унаследованный от pl.LightningModule.
  • Инициализируйте родительский класс, чтобы использовать все возможности LightningModule.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

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)
Редактировать и запускать код