Now try to increase the accuracy to at least 80% overall. You can:

• modify the architecture
• epochs
• learning rate
• momentum
• use early stopping

Include screenshots of

• the PyTorch model definiton
• the loss plot with the progress bar and selected model info
• the normalized confusion matrix and the accuracy info
• the prediction scores and class labels of 8 samples ('CNN: Predict' tab)

Do you see misclassified samples in the prediction scores? If yes, what can be said about them when looking at the top 3 probabilities that are listed?