schola.sb3.utils.save_model_as_onnx
- schola.sb3.utils.save_model_as_onnx(model, export_path)[source]
-
Save a stable baselines model as ONNX.
- Parameters:
-
-
model (stable_baselines3.common.base_class.BaseAlgorithm) – The model to save as ONNX.
-
export_path (str) – The path to save the model to.
-
- Return type:
-
None