Hi Luis!



 Luis_Costa:

First, the result of the softmax probability is always 1
                logits = model.forward(batch.to(device, dtype=torch.float)).cpu().detach()
                probabilities = F.softmax(logits, dim=1)
                print(probabilities)



Something is very fishy here.  I don’t believe it is possible to have
softmax() return all 1s.  (At least it shouldn’t be.)
Could you print out logits.shape, logits.min(), logits.max(),
probabilities.shape, probabilities.min(), and
probabilities.max()?
Also, what version of pytorch are you using?
Best.
K. Frank