Print Total Epochs during Test Step

Author: SuyashLakhotia

Diff for: train.py

@@ -103,8 +103,8 @@ def test_step(x_test, y_test, writer=None):
 
         time_str = datetime.datetime.now().isoformat()
         cur_epoch = step * batch_size / len(x_train)
-        print("{}: Step {}, Epoch {:.2f}, Loss {:g}, Accuracy {:g}".format(time_str, step, cur_epoch, loss,
-                                                                           accuracy))
+        print("{}: Step {}, Epoch {:.2f} / {}, Loss {:g}, Accuracy {:g}".format(time_str, step, cur_epoch,
+                                                                                num_epochs, loss, accuracy))
 
         summary = tf.Summary()
         summary.value.add(tag="loss_1", simple_value=loss)