fix default value. test=develop

dfb21219 · dzhwinter · a4cf2954 · dfb21219
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/tests/unittests/ir_memory_optimize_net_base.py ...ddle/fluid/tests/unittests/ir_memory_optimize_net_base.py +1 -1

未找到文件。
--- 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 : {}'.