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