We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 06a027f commit 95a5e7bCopy full SHA for 95a5e7b
official/vision/classification/shufflenet/model.py
@@ -110,7 +110,7 @@ def channel_shuffle(self, x):
110
111
112
class ShuffleNetV2(M.Module):
113
- def __init__(self, input_size=224, num_classes=1000, model_size="1.5x"):
+ def __init__(self, num_classes=1000, model_size="1.5x"):
114
super(ShuffleNetV2, self).__init__()
115
116
self.stage_repeats = [4, 8, 4]
0 commit comments