I am facing the same issue, with constant predictions. Were you able to solve this problem? If so, please help.
Also, I have another issue. I would like to use a pre-trained model to predict one of my own 28 classes.  So I suppose I should add a “Softmax” layer. I did this using add_module. However, the output from the model is still not with Softmax applied.
I have no experience with PyTorch or any other Deep Learning frameworks. Could someone please help?