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

View File

@ -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()