We just add evaluation of the training steps, note use of session.run to evaluate softmax of logits
ex05_tf_train.py
Model does not seem to train well. Issues without batch normalization: