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 7738bcee08e85700acd2c4eaec3d81d4c0130344..933253f2be45ced8229e22b86a17b0cb427eb930 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)