Skip to content

Commit 95a5e7b

Browse files
author
wangjianfeng
committed
fix(classification): remove unused args
1 parent 06a027f commit 95a5e7b

File tree

1 file changed

+1
-1
lines changed
  • official/vision/classification/shufflenet

1 file changed

+1
-1
lines changed

official/vision/classification/shufflenet/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ def channel_shuffle(self, x):
110110

111111

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

116116
self.stage_repeats = [4, 8, 4]

0 commit comments

Comments
 (0)