diff --git a/examples/image_classification/main.py b/examples/image_classification/main.py
index da47ca4d86e8cd5dfb50605fe1e00fbdcb137374..64396a6042f80cfbd53ff775ab95c41330894a9a 100644
--- a/examples/image_classification/main.py
+++ b/examples/image_classification/main.py
@@ -187,7 +187,7 @@ if __name__ == '__main__':
     parser.add_argument(
         "--resize-short-size",
         default=256,
-        type=float,
+        type=int,
         help="short size of keeping ratio resize")
     FLAGS = parser.parse_args()
     assert FLAGS.data, "error: must provide data path"