schola.sb3.utils.save_model_as_onnx

函数定义

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

将 stable baselines 模型保存为 ONNX。

参数

模型

类型： stable_baselines3.common.base_class.BaseAlgorithm
要保存为 ONNX 的模型。

导出路径

类型： str
保存模型的路径。

返回值

返回类型： None