merge into one function test=develop (#25184)

5 years ago · cb0472b05a
parent 68e93d8a17
commit cb0472b05a
1 changed files with 3 additions and 3 deletions
--- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_mobile_net.py
+++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_mobile_net.py
@ -517,10 +517,10 @@ class TestMobileNet(unittest.TestCase):
            np.allclose(dy_out, st_out),
            msg="dy_out: {}, st_out: {}".format(dy_out, st_out))

-    def test_mobileNetV1(self):
+    def test_mobileNet(self):
+        # MobileNet-V1
        self.assert_same_loss("MobileNetV1")
-
-    def test_mobileNetV2(self):
+        # MobileNet-V2
        self.assert_same_loss("MobileNetV2")