I think your model might output just one class (class0) and based on the class distribution in the current batch, you will get these results.
One mistake I just realized:
Remove the torch.exp in the return statement from the model.
NLLLoss expects log_softmax. Remove it and try it again.