schola.sb3.utils.convert_ckpt_to_onnx_for_unreal

Function Definition

schola.sb3.utils.convert_ckpt_to_onnx_for_unreal(
    trainer,
    model_path='./ckpt/ppo_final.zip',
    export_path='./ckpt/OnnxFiles/Model.onnx'
)

Convert a stable baselines model to ONNX for use in Unreal.

Parameters

trainer

Type: stable_baselines3.common.base_class.BaseAlgorithm
The trainer to load the model from.

model_path

Type: str
Default: './ckpt/ppo_final.zip'
The path to the model to convert.

export_path

Type: str
Default: './ckpt/OnnxFiles/Model.onnx'
The path to save the converted model to.

Returns

Return type: None