@ptrblck
The accuracy improved as I changed predicted=torch.argmax(outputs,1) in the test part of the code…right now it is showing about 97% test accuracy. Thank you very much sir…Although why torch.max(outputs,1) was causing trouble is a mystery…I would like to know about it