From 6deac40724995e04039f1fda19b7ea037bf1597c Mon Sep 17 00:00:00 2001 From: dzhwinter Date: Mon, 18 Feb 2019 00:41:26 +0800 Subject: [PATCH] polish code for reading. test=develop --- .../fluid/tests/unittests/test_ir_memory_optimize_transformer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/paddle/fluid/tests/unittests/test_ir_memory_optimize_transformer.py b/python/paddle/fluid/tests/unittests/test_ir_memory_optimize_transformer.py index f32e1161a..c0f480e34 100644 --- a/python/paddle/fluid/tests/unittests/test_ir_memory_optimize_transformer.py +++ b/python/paddle/fluid/tests/unittests/test_ir_memory_optimize_transformer.py @@ -22,6 +22,7 @@ os.environ[ 'RECORDIO_FILENAME'] = '/tmp/ir_memory_optimize_transformer.wmt16.recordio' from test_parallel_executor_transformer import TestTransformer +from test_parallel_executor_transformer import transformer # NOTE(dzhwinter): test diferent strategy colisions. -- GitLab