diff --git a/official/vision/classification/shufflenet/model.py b/official/vision/classification/shufflenet/model.py
index 68d05d7e4ef0dda4411da527b47e955426093aa0..7622a7c17ba631002797430653d1d4da531a5215 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]