schola.sb3.utils.save_model_as_onnx

Function Definition

schola.sb3.utils.save_model_as_onnx(model, export_path)

Save a stable baselines model as ONNX.

Parameters

model

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

export_path

Type: str
The path to save the model to.

Returns

Return type: None