ПочатиПочніть безкоштовно

Модель із двома входами

Дані готові — час побудувати архітектуру моделі з двома входами! Для цього ви створите клас моделі з такими методами:

  • .__init__(), у якому ви визначите підмережі, згрупувавши шари; тут ви задасте два шари для обробки двох входів і класифікатор, що повертає оцінку належності до кожного класу.

  • forward(), у якому ви пропустите обидва входи через відповідні заздалегідь визначені підмережі, конкатенуєте їхні виходи та передасте результат до класифікатора.

torch.nn уже імпортовано для вас як nn. Рушаймо!

Ця вправа є частиною курсу

Середній рівень Deep Learning з PyTorch

Переглянути курс

Інтерактивна практична вправа

Спробуйте виконати цю вправу, доповнивши цей зразок коду.

class Net(nn.Module):
    def __init__(self):
        super().__init__()
        # Define sub-networks as sequential models
        ____ = ____(
            nn.Conv2d(1, 16, kernel_size=3, padding=1),
            nn.MaxPool2d(kernel_size=2),
            nn.ELU(),
            nn.Flatten(),
            nn.Linear(16*32*32, 128)
        )
        ____ = ____(
            nn.Linear(30, 8),
            nn.ELU(), 
        )
        ____ = ____(
            nn.Linear(128 + 8, 964), 
        )
Редагувати та запускати код