utils.convert_ckpt_to_onnx_for_unreal

Full path: schola.sb3.utils.convert_ckpt_to_onnx_for_unreal

schola.sb3.utils.convert_ckpt_to_onnx_for_unreal

convert_ckpt_to_onnx_for_unreal

convert_ckpt_to_onnx_for_unreal(trainer=<MagicMock name='mock.PPO' id='2386012941888'>, 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 (stable_baselines3.common.base_class.BaseAlgorithm) : The trainer to load the model from.

model_path (str) : The path to the model to convert.

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

Returns

None

---