Update stable_baselines3_example.py

edbeeching · Aug 22, 2024 · 1df043d · 1df043d
1 parent 7806f26
commit 1df043d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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():