From f0d4938bf7dc2802bb8e40b3d59b9d864b48c934 Mon Sep 17 00:00:00 2001
From: LielinJiang <jianglielin@baidu.com>
Date: Wed, 22 Apr 2020 04:05:15 +0000
Subject: [PATCH] fix typo

---
 examples/image_classification/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/image_classification/main.py b/examples/image_classification/main.py
index da47ca4..64396a6 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"
-- 
GitLab