diff --git a/test/test_test_mnist.py b/test/test_test_mnist.py index dd6adce4515..92992f8e73b 100644 --- a/test/test_test_mnist.py +++ b/test/test_test_mnist.py @@ -245,7 +245,7 @@ def test_loop_fn(loader): # train_loop_fn(train_device_loader, epoch) # xm.master_print('Epoch {} train end {}'.format(epoch, test_utils.now())) accuracy = test_loop_fn(test_device_loader) - # xm.master_print('Epoch {} test end {}, Accuracy={:.2f}'.format(epoch, test_utils.now(), accuracy)) + xm.master_print('Epoch {} test end {}, Accuracy={:.2f}'.format(epoch, test_utils.now(), accuracy)) max_accuracy = max(accuracy, max_accuracy) # test_utils.write_to_summary(writer, epoch, dict_to_write={'Accuracy/test': accuracy}, write_xla_metrics=True) # if flags.metrics_debug: xm.master_print(met.metrics_report())