Hi Akib!



 akib62:

I have a CNN model to classify the MNIST digits problem.


Perhaps some of the posts in the following thread could be helpful:

  
    
    
    How to get continuous value from output layer vision
  
  
    I have a code (from here) to classify the MINST digits. The code is working fine. Here they used CrossEntropyLoss and Adam optimizer. 
The model code is given below 
class CNN(nn.Module):
    def __init__(self):
        super(CNN, self).__init__()
        self.conv1 = nn.Sequential(         
            nn.Conv2d(
                in_channels=1,              
                out_channels=16,            
                kernel_size=5,              
                stride=1,                   
    …
  


Best.
K. Frank