Skip to content

schola.sb3.utils.save_model_as_onnx

schola.sb3.utils.save_model_as_onnx(model, export_path) : 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