Allow computing misclassification as part of training stats
Improve error computation in training stats.
Fix softmax and cross-entropy (for numeric stability)