I am getting this error when I am running my training loop
 TypeError: conv2d(): argument 'input' (position 1) must be Tensor, not int
This is a simple CNN model, here is the model arch:
class ResNet18(nn.Module):

    def __init__(self):

        super(ResNet18, self).__init__()

        self.network = nn.Sequential(

            nn.Conv2d(3, 16, kernel_size=(3,3), stride=1),

            nn.ReLU(),

            nn.MaxPool2d(kernel_size=(2,2)),

            

            nn.Conv2d(16, 32, kernel_size=(3,3), stride=1),

            nn.ReLU(),

            nn.MaxPool2d(kernel_size=(2,2)),

                        

            nn.Flatten(),

            nn.Linear(16*16*32, 64),

            nn.ReLU(),

            nn.Linear(64, 32),

            nn.ReLU(),

            nn.Linear(32, 7),

            )

        

    def forward(self, x):

        return self.network(x)

Can someone help?
P.S: I know I called this model ResNet18 even though it’s not lol