diff --git a/ppcls/modeling/architectures/shufflenet_v2.py b/ppcls/modeling/architectures/shufflenet_v2.py index f6ffd29f9ac0b6574f5393b5170faf0115caaf36..a051e440d27c0ca434c59e96e1d4faa3bd3e715b 100644 --- a/ppcls/modeling/architectures/shufflenet_v2.py +++ b/ppcls/modeling/architectures/shufflenet_v2.py @@ -337,5 +337,5 @@ def ShuffleNetV2_x2_0(**args): def ShuffleNetV2_swish(**args): - model = ShuffleNet(scale=1.0, act=swish, **args) + model = ShuffleNet(scale=1.0, act="swish", **args) return model