diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_ptb_lm.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_ptb_lm.py index ea0529ffb28d46e16a7731516bcbf228aa81cf21..7b7ff66343a40fb09ed6eae3e297b6b793d34d42 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_ptb_lm.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_ptb_lm.py @@ -314,4 +314,6 @@ class TestPtb(unittest.TestCase): if __name__ == '__main__': - unittest.main() + # switch into new eager mode + with fluid.framework._test_eager_guard(): + unittest.main() diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet.py index 2fbb933f9ced472720c3e3d7b137424ec2c5ee3b..efb69b530efc96d49f99f0117a947efef8ff169c 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet.py @@ -366,4 +366,6 @@ class TestResnet(unittest.TestCase): if __name__ == '__main__': - unittest.main() + # switch into new eager mode + with fluid.framework._test_eager_guard(): + unittest.main() diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_se_resnet.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_se_resnet.py index eb17264977f508599adbfc6d2c2fa9c773c58534..eba7e238bb5906359bd7ee3734bb3b73bb73feda 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_se_resnet.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_se_resnet.py @@ -505,4 +505,6 @@ class TestSeResnet(unittest.TestCase): if __name__ == '__main__': - unittest.main() + # switch into new eager mode + with fluid.framework._test_eager_guard(): + unittest.main() diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_seq2seq.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_seq2seq.py index 3c0c046191c9da482645ee93d15650e077d77b3b..a363526530d11daebe46e98154ad87b355f339b6 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_seq2seq.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_seq2seq.py @@ -203,4 +203,6 @@ class TestSeq2seq(unittest.TestCase): if __name__ == '__main__': - unittest.main() + # switch into new eager mode + with fluid.framework._test_eager_guard(): + unittest.main() diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_simnet.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_simnet.py index 01e9ed07efa83d35baaed15550cc9d7257337ba1..ce88ea74af23ef7b3f0ba142dcd61b4b29ec0634 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_simnet.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_simnet.py @@ -170,4 +170,6 @@ class TestSimnet(unittest.TestCase): if __name__ == '__main__': - unittest.main() + # switch into new eager mode + with fluid.framework._test_eager_guard(): + unittest.main() diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_transformer.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_transformer.py index 451ceea75c09436139f0a0bad57ff80b0127591f..06f2c60dfae9f69b2dd186c064c756021021acb5 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_transformer.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_transformer.py @@ -402,4 +402,6 @@ class TestTransformer(unittest.TestCase): if __name__ == '__main__': - unittest.main() + # switch into new eager mode + with fluid.framework._test_eager_guard(): + unittest.main() diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_tsm.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_tsm.py index 7ca0a1a539e38299ef81d4a58f82c4a4dfd1e5fb..37fc78458dd789a3fc4d32bbf212be1b5aaf700a 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_tsm.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_tsm.py @@ -347,4 +347,6 @@ class TestTsm(unittest.TestCase): if __name__ == '__main__': - unittest.main() + # switch into new eager mode + with fluid.framework._test_eager_guard(): + unittest.main()