fix(classification): remove unused args

95a5e7bd · wangjianfeng · 06a027f8 · 95a5e7bd
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

official/vision/classification/shufflenet/model.py official/vision/classification/shufflenet/model.py +1 -1

未找到文件。
--- a/official/vision/classification/shufflenet/model.py
+++ b/official/vision/classification/shufflenet/model.py
@@ -110,7 +110,7 @@ class ShuffleV2Block(M.Module):


 class ShuffleNetV2(M.Module):
-    def __init__(self, input_size=224, num_classes=1000, model_size="1.5x"):
+    def __init__(self, num_classes=1000, model_size="1.5x"):
        super(ShuffleNetV2, self).__init__()

        self.stage_repeats = [4, 8, 4]