class cnn_model(nn.Module):
def init(self):
super(cnn_model,self).init()
self.cov1 = nn.Conv2d(1,16,kernel_size=5)
self.cov2 = nn.Conv2d(16,32, kernel_size = 5)
self.fc = nn.Linear(1568,10)
self.relu = nn.ReLU()
self.Maxpool = nn.MaxPool2d(kernel_size = 2)
def forward(self, x):
    n_size = x.size(0)
    out = self.Maxpool(self.relu(self.cov1(x)))
    out = self.Maxpool(self.relu(self.cov2(out)))
    out = out.view(n_size,-1)
    out = self.fc(out)
    
    return out