From 95a5e7bdba16f07209aad00178f2a169fbc553fc Mon Sep 17 00:00:00 2001 From: wangjianfeng Date: Mon, 27 Apr 2020 20:01:32 +0800 Subject: [PATCH] fix(classification): remove unused args --- official/vision/classification/shufflenet/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/official/vision/classification/shufflenet/model.py b/official/vision/classification/shufflenet/model.py index 68d05d7..7622a7c 100644 --- 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] -- GitLab