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)