Skip to content

schola.sb3.utils.convert_ckpt_to_onnx_for_unreal

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 (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