polish code for reading. test=develop

7 years ago · d0a2a202d0
parent c1455e606d
commit d0a2a202d0
1 changed files with 3 additions and 2 deletions
--- 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"
+import paddle.fluid as fluid
-os.environ['FLAGS_fast_eager_deletion_mode'] = True
+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'