diff --git a/utils/RunONNXModel.py b/utils/RunONNXModel.py index 24ab67220a..56c2b603cb 100755 --- a/utils/RunONNXModel.py +++ b/utils/RunONNXModel.py @@ -160,6 +160,16 @@ def check_non_negative(argname, value): "Otherwise, compile model and save it into the given path.", ) +parser.add_argument( + "-o", + "--default-model-name", + metavar="MODEL_NAME", + type=str, + default="model", + help="Change the default model name that is used for two generated files: " + " .so and .constants.bin. Default is model.", +) + parser.add_argument( "--save-ref", metavar="PATH", @@ -645,7 +655,7 @@ def __init__(self, model_file, **kwargs): # Default model name that will be used for the compiled model. # e.g. model.so, model.constants.bin, ... - self.default_model_name = "model" + self.default_model_name = args.default_model_name # Handle cache_model. if args.cache_model: