polish code for reading. test=develop

d0a2a202 · dzhwinter · c1455e60 · d0a2a202
隐藏空白更改
内联并排

Showing with 3 addition and 2 deletion

python/paddle/fluid/tests/unittests/test_ir_memory_optimize_transformer.py ...id/tests/unittests/test_ir_memory_optimize_transformer.py +3 -2

未找到文件。
--- a/python/paddle/fluid/tests/unittests/test_ir_memory_optimize_transformer.py
+++ b/python/paddle/fluid/tests/unittests/test_ir_memory_optimize_transformer.py
@@ -14,9 +14,10 @@

 import os
 import unittest
-os.environ['FLAGS_eager_delete_tensor_gb'] = "0.0"
-os.environ['FLAGS_fast_eager_deletion_mode'] = True
+import paddle.fluid as fluid
+import paddle.fluid.core as core

+os.environ['FLAGS_eager_delete_tensor_gb'] = "0.0"
 os.environ[
    'RECORDIO_FILENAME'] = '/tmp/ir_memory_optimize_transformer.wmt16.recordio'