diff --git a/python/paddle/fluid/tests/unittests/test_partial_eager_deletion_transformer.py b/python/paddle/fluid/tests/unittests/test_partial_eager_deletion_transformer.py index fc1d762ec9296e38169730a4373027017875ff6e..7607189454b2264523176b6853fd9debddf47eed 100644 --- a/python/paddle/fluid/tests/unittests/test_partial_eager_deletion_transformer.py +++ b/python/paddle/fluid/tests/unittests/test_partial_eager_deletion_transformer.py @@ -17,8 +17,7 @@ import unittest os.environ['FLAGS_eager_delete_tensor_gb'] = "0.0" os.environ['FLAGS_memory_fraction_of_eager_deletion'] = "0.55" -os.environ[ - 'RECORDIO_FILENAME'] = './partial_eager_deletion_transformer.wmt16.recordio' +os.environ['RECORDIO_FILENAME'] = './p_gc_transformer.wmt16.recordio' from test_parallel_executor_transformer import TestTransformer