sb3_to_onnx.py 脚本

此脚本将 Stable Baselines3 模型转换为 ONNX 模型，以便在 Unreal Engine 中使用。

用法

usage: sb3_to_onnx [-h] [--model-path MODEL_PATH] [--output-path OUTPUT_PATH]

可选参数

--model-path

要转换的 Stable Baselines3 模型的路径

• 默认: './ckpt/ppo_final.zip'
• 类型: str
• 必需: 否

--output-path

保存 ONNX 模型的位置

• 默认: './ckpt/ppo_final/model.onnx'
• 类型: str
• 必需: 否