fix default value. test=develop

6 years ago · dfb2121967
parent a4cf295471
commit dfb2121967
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/tests/unittests/ir_memory_optimize_net_base.py
+++ b/python/paddle/fluid/tests/unittests/ir_memory_optimize_net_base.py
@ -121,7 +121,7 @@ class TestIrMemOptBase(BuildIrMemOptBase):
            return
        baseline_first_loss, baseline_last_loss = None, None
-        for use_cuda in [True, False]:
+        for use_cuda in [True]:
            for use_python_mem_opt in [True, False]:
                print(
                    'network: {}, use_cuda: {}, use_python_mem_opt: {}, use_ir_mem_opt : {}'.