diff --git a/onnx_scripts/profile_onnx.py b/onnx_scripts/profile_onnx.py index ad18c7f..adb593b 100644 --- a/onnx_scripts/profile_onnx.py +++ b/onnx_scripts/profile_onnx.py @@ -37,5 +37,5 @@ def profile_onnx_model(path_to_onnx: str, num_runs: int = 100): if __name__ == "__main__": settings = get_app_settings() - output_path = os.path.join("onnx_files", "convert_to_onnx.onnx") + output_path = os.path.join("onnx_files", "inference_recognition_model.onnx") profile_onnx_model(output_path)