We just add evaluation of the training steps, note use of session.run to evaluate softmax of logits

code:

ex05_tf_train.py

Notes:

Model does not seem to train well. Issues without batch normalization: