data_transforms = {

‘train’: transforms.Compose([

transforms.RandomResizedCrop(224),

transforms.RandomHorizontalFlip(),

transforms.ToTensor(),

transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])

]),

‘val’: transforms.Compose([

transforms.Resize(224),

transforms.CenterCrop(256),

transforms.ToTensor(),

transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])

]),

}

class Net(nn.Module):

def **init**(self):

super(Net, self).**init**()

self.conv1 = nn.Conv2d(3, 64, kernel_size=6, padding=2)

self.pool = nn.MaxPool2d(2, 2)

self.conv2 = nn.Conv2d(64, 128, kernel_size=6, padding=2)

self.fc1 = nn.Linear(256 * 6 * 6, 4096)

self.fc2 = nn.Linear(4096, 4096)

self.fc3 = nn.Linear(4096, 2)

```
def forward(self, x):
x = self.pool(F.relu(self.conv1(x)))
x = self.pool(F.relu(self.conv2(x)))
x = x.reshape(x.size(0), -1)
x = F.relu(self.fc1(x))
x = F.relu(self.fc2(x))
x = self.fc3(x)
return x
```

RuntimeError: size mismatch, m1: [4 x 387200], m2: [9216 x 4096]

how can I set parameters for this model?