diff --git a/ppcls/modeling/architectures/shufflenet_v2.py b/ppcls/modeling/architectures/shufflenet_v2.py index 54a5a81ec..f6ffd29f9 100644 --- a/ppcls/modeling/architectures/shufflenet_v2.py +++ b/ppcls/modeling/architectures/shufflenet_v2.py @@ -307,7 +307,7 @@ class ShuffleNet(Layer): def ShuffleNetV2_x0_25(**args): - model = ShuffleNetV2(scale=0.25, **args) + model = ShuffleNet(scale=0.25, **args) return model