Skip to content

rllib_to_onnx.py Script

This script converts a RLlib model to an ONNX model for use in Unreal Engine.

Usage

Terminal window
usage: rllib_to_onnx [-h] [--policy-checkpoint-path POLICY_CHECKPOINT_PATH] [--output-path OUTPUT_PATH]

Optional Arguments

--policy-checkpoint-path

Path to the RLlib policy checkpoint to convert

  • Type: str
  • Required: False

--output-path

Path where the ONNX model will be saved

  • Type: str
  • Required: False