Its 25/436/404 without augmentation. But its visually a pretty easy task and thats why I get such a good accuracy on the prediction I think.
You think a weight tensor like [0.8,0.1,0.1] could work?
Or do you have any other suggestion how I can reach higher probabilities for the predictions?