Hi,
Can you report probabilities for a sample test/train image?
And also, Could you please explain why are you using log(softmax)? For comparison purposes or in your case, softmax is a more interpretable choice. Softmax maps input to probability space in range of [0, 1] so applying log on top of it, will lead to a all negative numbers but sum of them won’t be 1.
Bests