ray.rllib.policy.torch_policy_v2.TorchPolicyV2.导出模型

TorchPolicyV2.export_model(export_dir: str, onnx: int | None = None) → None

将策略的模型导出到本地目录以供服务。

创建一个 TorchScript 模型并保存它。

参数:

• export_dir – 本地可写目录或文件名。

• onnx – 如果提供，将导出 ONNX 格式的模型。此参数的值设置要使用的 ONNX OpSet 版本。