Для torch.nn.Module ()
Согласно официальной документации: Базовый класс для всех нейросетевых модулей. Ваши модели также должны быть подклассами этого класса. Модули также могут содержать другие Модули, позволяющие вкладывать их в древовидную структуру. Вы можете назначить подмодули как обычные атрибуты.
import torch.nn as nn
import torch.nn.functional as F
class Model(nn.Module):
def __init__(self):
super(Model, self).__init__()
self.conv1 = nn.Conv2d(1, 20, 5)
self.conv2 = nn.Conv2d(20, 20, 5)
def forward(self, x):
x = F.relu(self.conv1(x))
return F.relu(self.conv2(x))
Использовано super(Model, self).__init__()
Почему бы и нет super().__init__(Model, self)