utils.save_model_as_onnx

Full path: schola.sb3.utils.save_model_as_onnx

schola.sb3.utils.save_model_as_onnx

save_model_as_onnx(model, export_path)

Save a stable baselines model as ONNX.

model (stable_baselines3.common.base_class.BaseAlgorithm) : The model to save as ONNX.

export_path (str) : The path to save the model to.

None