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