Underwhelming performance in neural network with PyTorch

I would use a framework like fastai with models from timm.

IIRC you should probably be able to get much better results with “3 lines of code”. For more info check the first lesson in the course for details

2 Likes