mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update export_model.py

pull/369/head
littletomatodonkey 2020-11-04 22:49:20 +08:00 committed by GitHub
parent c3922941de
commit d4b1f6aff3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/export_model.py
View File

@ -36,7 +36,7 @@ def parse_args():
parser.add_argument( parser.add_argument(
"-o", "--output_path", type=str, default="./inference/cls_infer") "-o", "--output_path", type=str, default="./inference/cls_infer")
parser.add_argument("--class_dim", type=int, default=1000) 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) # parser.add_argument("--img_size", type=int, default=224)
return parser.parse_args() return parser.parse_args()