From 9556beafc455476a9f1601d11eacbc3dea737b7e Mon Sep 17 00:00:00 2001
From: WangZhen <23097963+0x45f@users.noreply.github.com>
Date: Fri, 28 Jul 2023 10:46:37 +0800
Subject: [PATCH] Fix test_resnet and test_resnet_v2 ut (#55723)

* Fix test_resnet and test_resnet_v2 ut

* Remove ut
---
 test/dygraph_to_static/CMakeLists.txt |  4 ++--
 test/dygraph_to_static/test_resnet.py | 14 --------------
 2 files changed, 2 insertions(+), 16 deletions(-)

diff --git a/test/dygraph_to_static/CMakeLists.txt b/test/dygraph_to_static/CMakeLists.txt
index d8bec02ae9f..3b1d7f726b1 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 9fcc1a803f8..afb001358b4 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)
-- 
GitLab