schola.sb3.utils.convert_ckpt_to_onnx_for_unreal
- schola.sb3.utils.convert_ckpt_to_onnx_for_unreal(trainer=<class ‘stable_baselines3.ppo.ppo.PPO’>, model_path=’./ckpt/ppo_final.zip’, export_path=’./ckpt/OnnxFiles/Model.onnx’)[source]
-
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.
-
- Return type:
-
None