Thanks for your reply.
It means I’m training a model to classify image is fake or real.
My last layer of model is nn.Linear before I fit it to softmax.
I’m training and looking like the model converged well due to loss, acc decrease gradually.
In test pharse, I print output after linear function as above image.
The output is always ~ U[-k, k] distribution (like above ‘’’self.prob tensor([[ 4.2354, -4.1672]] ‘’’)
I’m new in training model, so If my question is tough to understand, pardon me for that. I’m always highly appreciated your help.