What I need to change the CNN model for a regression problem

[KFrank]

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

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