rllib_to_onnx.py Script

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

Usage

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