未验证 提交 d4b1f6af 编写于 作者: L littletomatodonkey 提交者: GitHub

Update export_model.py

上级 c3922941
......@@ -36,7 +36,7 @@ def parse_args():
parser.add_argument(
"-o", "--output_path", type=str, default="./inference/cls_infer")
parser.add_argument("--class_dim", type=int, default=1000)
parser.add_argument("--load_static_weights", type=str2bool, default=True)
parser.add_argument("--load_static_weights", type=str2bool, default=False)
# parser.add_argument("--img_size", type=int, default=224)
return parser.parse_args()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册