1. Learn
    2. /
  2. Курси
    3. /
  3. Глибоке навчання для зображень із PyTorch
Connected

Вправа

Створення послідовного блока

Ви вирішили переробити свій шаблон бінарної моделі CNN, створивши блок згорткових шарів. Це допоможе послідовно поєднувати кілька шарів. З оновленою моделлю ви зможете легко конструювати різні архітектури CNN.

torch і torch.nn як nn вже імпортовано.

Інструкції

100 XP
  • У методі __init__() визначте блок згорткових шарів і присвойте його до self.conv_block.
  • У проході forward() пропустіть вхідні дані через визначений вами згортковий блок.