down into cpp

pytorch · Apr 25, 2024 · 6afbc66 · 6afbc66
1 parent c7a2d9a
commit 6afbc66
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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())