diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_mnist_pure_fp16.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_mnist_pure_fp16.py index 4ddc9d1aa0860996a92fcef0ba4604339ae8b59b..1860362896cfb1c161891e9bc2dcbbd87faf9468 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_mnist_pure_fp16.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_mnist_pure_fp16.py @@ -91,13 +91,13 @@ class TestPureFP16(TestMNIST): loss_data.append(avg_loss.numpy()[0]) # save checkpoint mnist.clear_gradients() - if batch_id % 10 == 0: + if batch_id % 2 == 0: print( "Loss at epoch {} step {}: loss: {:}, acc: {}, cost: {}" .format(epoch, batch_id, avg_loss.numpy(), acc.numpy(), time() - start)) start = time() - if batch_id == 50: + if batch_id == 10: break return loss_data