diff --git 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 index 88e57e206143ab88932160303bd284f312cb1edf..49002ed57da23417a7d497dbadab3c178b791425 100644 --- 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")