Need to recognize digits

vision
1

I want to recognize digits on images, but no luck for now. Although images seem to be quite recognizable NN does not seem to do any better than random guess.
Notebook:

Images:
https://drive.google.com/drive/folders/12nhWJo73gHUXWa0vg38cCOlz_aKHesKN?usp=sharing

Any help is appreciated.
Thank for your time!

2

Your model uses:

nn.Sigmoid(),
nn.Softmax()

as the last layers which is wrong.
nn.CrossEntropyLoss expects raw logits so remove both layers and rerun your script.