diff --git a/test/dygraph_to_static/CMakeLists.txt b/test/dygraph_to_static/CMakeLists.txt index d8bec02ae9ff81aaf98b07a4e32089046dd2fc72..3b1d7f726b1e83aa2df02759f14a782cb570e66d 100644 --- a/test/dygraph_to_static/CMakeLists.txt +++ b/test/dygraph_to_static/CMakeLists.txt @@ -75,9 +75,9 @@ set_tests_properties(test_bmn PROPERTIES TIMEOUT 120) set_tests_properties(test_build_strategy PROPERTIES TIMEOUT 120) if(NOT WIN32) - set_tests_properties(test_resnet_v2 PROPERTIES TIMEOUT 120) + set_tests_properties(test_resnet_v2 PROPERTIES TIMEOUT 180) set_tests_properties(test_tsm PROPERTIES TIMEOUT 900) - #set_tests_properties(test_resnet PROPERTIES TIMEOUT 120) + set_tests_properties(test_resnet PROPERTIES TIMEOUT 240) endif() if(APPLE) diff --git a/test/dygraph_to_static/test_resnet.py b/test/dygraph_to_static/test_resnet.py index 9fcc1a803f8228499676ae22275924d17c486e25..afb001358b48c269c6c5326d99d7446e9c58bda7 100644 --- a/test/dygraph_to_static/test_resnet.py +++ b/test/dygraph_to_static/test_resnet.py @@ -426,20 +426,6 @@ class TestResnet(unittest.TestCase): ) self.verify_predict() - def test_resnet_composite_backward(self): - core._set_prim_backward_enabled(True) - static_loss = self.train(to_static=True) - core._set_prim_backward_enabled(False) - dygraph_loss = self.train(to_static=True) - np.testing.assert_allclose( - static_loss, - dygraph_loss, - rtol=1e-05, - err_msg='static_loss: {} \n dygraph_loss: {}'.format( - static_loss, dygraph_loss - ), - ) - def test_resnet_composite_forward_backward(self): core._set_prim_all_enabled(True) static_loss = self.train(to_static=True)