This is my model:
class SCNN(nn.Module):
def init(self):
super(SCNN, self).init()
    self.conv1 = nn.Conv2d(in_channels=3, out_channels=64, kernel_size=3, stride=1, padding=1, bias=False)

    self.conv2 = nn.Conv2d(in_channels=64, out_channels=128, kernel_size=3, stride=1, padding=1, bias=False)

    self.conv3 = nn.Conv2d(in_channels=128, out_channels=128, kernel_size=3, stride=1, padding=1, bias=False)

    self.fc1 = nn.Linear(128, 1024, bias=False)

    self.fc2 = nn.Linear(1024, 10)