diff --git a/tools/export_model.py b/tools/export_model.py index 46a8a8b8dfb8dd4c33931007692aa91376530089..b6c03efba7b48f23ffb8794996972c79bee76c57 100755 --- a/tools/export_model.py +++ b/tools/export_model.py @@ -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(