Skip to content

GitLab

"docs/source/vscode:/vscode.git/clone" did not exist on "2db73f4a509864d38c668979c58ce1cee57a84db"
Commit 0ff2aef2 authored by WenmuZhou's avatar WenmuZhou
Browse files

rename inference model save path

parent 4fd696cc
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
tools/export_model.py
View file @ 0ff2aef2
......@@ -57,8 +57,7 @@ def main():
init_model(config, model, logger)
model.eval()
save_path = '{}/{}/inference'.format(FLAGS.output_path,
config['Architecture']['model_type'])
save_path = '{}/inference'.format(FLAGS.output_path)
infer_shape = [3, 32, 100] if config['Architecture'][
'model_type'] != "det" else [3, 640, 640]
model = to_static(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment