跳至内容

rllib_to_onnx.py 脚本

此脚本将 RLlib 模型转换为 ONNX 模型,以便在 Unreal Engine 中使用。

用法

终端窗口
usage: rllib_to_onnx [-h] [--policy-checkpoint-path POLICY_CHECKPOINT_PATH] [--output-path OUTPUT_PATH]

可选参数

--policy-checkpoint-path

要转换的 RLlib 策略检查点的路径

  • 类型: str
  • 必需:

--output-path

保存 ONNX 模型的路径

  • 类型: str
  • 必需:
© . This site is unofficial and not affiliated with AMD.