提交 95a5e7bd 编写于 作者: W wangjianfeng

fix(classification): remove unused args

上级 06a027f8
...@@ -110,7 +110,7 @@ class ShuffleV2Block(M.Module): ...@@ -110,7 +110,7 @@ class ShuffleV2Block(M.Module):
class ShuffleNetV2(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__() super(ShuffleNetV2, self).__init__()
self.stage_repeats = [4, 8, 4] self.stage_repeats = [4, 8, 4]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册