From 1df043d94276229f24181249d7f50fc1d70268b7 Mon Sep 17 00:00:00 2001 From: Ivan-267 <61947090+Ivan-267@users.noreply.github.com> Date: Thu, 22 Aug 2024 20:45:23 +0200 Subject: [PATCH] Update stable_baselines3_example.py --- examples/stable_baselines3_example.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/stable_baselines3_example.py b/examples/stable_baselines3_example.py index 8f94a14..630f204 100644 --- a/examples/stable_baselines3_example.py +++ b/examples/stable_baselines3_example.py @@ -115,7 +115,7 @@ def handle_onnx_export(): if args.onnx_export_path is not None: path_onnx = pathlib.Path(args.onnx_export_path).with_suffix(".onnx") print("Exporting onnx to: " + os.path.abspath(path_onnx)) - export_model_as_onnx(model, str(path_onnx), True) + export_model_as_onnx(model, str(path_onnx)) def handle_model_save():