Hello! Im new using pytorch and I cant manage to find a proper solution to this error, could anybody help me?
class MaskCnnModel(ImageClassificationBase):
def init(self):
super().init()
self.network = nn.Sequential(
nn.Conv2d(3, 86, kernel_size=3, padding=1),
nn.ReLU(),
nn.Conv2d(86, 172, kernel_size=3, stride=1, padding=1),
nn.ReLU(),
nn.MaxPool2d(2, 2), # output: 172 x 43 x 43
nn.Conv2d(172, 258, kernel_size=3, stride=1, padding=1),
nn.ReLU(),
nn.Conv2d(258, 258, kernel_size=3, stride=1, padding=1),
nn.ReLU(),
nn.MaxPool2d(2, 2), # output: 344 x 21.5 x 21.5
nn.Conv2d(258, 344, kernel_size=3, stride=1, padding=1),
nn.ReLU(),
nn.Conv2d(344, 344, kernel_size=3, stride=1, padding=1),
nn.ReLU(),
nn.MaxPool2d(2, 2), # output: 688 x 10.8 x 10.8
nn.Flatten(),
nn.Linear(344*11*11, 1376),
nn.ReLU(),
nn.Linear(1376, 688),
nn.ReLU(),
nn.Linear(688, 2))
def forward(self, xb):
return self.network(xb)