diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_cycle_gan.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_cycle_gan.py index d6840ed62810e13648b869340a41691aa0e89101..8a9a1e19205fbfac264dc99f0dc2f9fa2932fd48 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_cycle_gan.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_cycle_gan.py @@ -631,4 +631,5 @@ class TestCycleGANModel(unittest.TestCase): if __name__ == "__main__": - unittest.main() + with fluid.framework._test_eager_guard(): + unittest.main() diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_lac.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_lac.py index 88c6060abf7d9d05d7c975b5942cb11e884ffd52..19965821e8750d68546ba6865f596391b29f5c4d 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_lac.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_lac.py @@ -610,4 +610,5 @@ class TestLACModel(unittest.TestCase): if __name__ == "__main__": - unittest.main() + with fluid.framework._test_eager_guard(): + unittest.main() diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet_amp.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet_amp.py index 509accd8f8ab256ebaa76f25d381512c763debe5..1d45e906cd378f92925dca003ebf34c0b129a2d9 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet_amp.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet_amp.py @@ -116,4 +116,5 @@ class TestResnet(unittest.TestCase): if __name__ == '__main__': - unittest.main() + with fluid.framework._test_eager_guard(): + unittest.main() diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet_pure_fp16.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet_pure_fp16.py index cf5c2b731141fa993f81de0f957b828178a01760..49d114730e4ed94441a1c5a3e5ad187dba9e1c2d 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet_pure_fp16.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet_pure_fp16.py @@ -125,4 +125,5 @@ class TestResnet(unittest.TestCase): if __name__ == '__main__': - unittest.main() + with fluid.framework._test_eager_guard(): + unittest.main()