跳至内容

schola.scripts.ray.settings.RLlibScriptArgs

类定义

class schola.scripts.ray.settings.RLlibScriptArgs(enable_checkpoints=False, checkpoint_dir='./ckpt', save_freq=100000, name_prefix_override=None, export_onnx=False, save_final_policy=False, launch_unreal=False, port=None, executable_path=None, headless=False, map=None, fps=None, disable_script=False, training_settings=, algorithm_settings=, logging_settings=, resume_settings=, network_architecture_settings=, resource_settings=, plugins=)

基类: ScriptArgs

RLlib 脚本参数的顶层数据类。此类汇总了配置 RLlib 训练过程所需的所有设置。它包括训练、算法、日志记录、从检查点恢复、网络架构和资源分配的设置。这允许以结构化的方式全面配置 RLlib 训练作业。

参数

启用检查点

类型: bool

检查点目录

类型: str

保存频率

Type: int

覆盖名称前缀

类型: str | None

导出为 ONNX

类型: bool

保存最终策略

类型: bool

启动 Unreal

类型: bool

port

类型: int | None

executable_path

类型: str | None

无头模式

类型: bool

map

类型: str | None

帧率

类型: int | None

disable_script

类型: bool

训练设置

类型: TrainingSettings

算法设置

类型: PPOSettings | APPOSettings | IMPALASettings

日志设置

类型: LoggingSettings

恢复设置

类型: ResumeSettings

网络架构设置

类型: NetworkArchitectureSettings

资源设置

类型: ResourceSettings

插件

类型: List[RLLibLauncherExtension]

属性

算法设置

类型: PPOSettings | APPOSettings | IMPALASettings

检查点目录

启用保存检查点。

disable_script

表示是否禁用 Unreal Engine Schola 插件中的自动启动脚本设置的标志。

启用检查点

启用保存检查点

executable_path

独立环境启动时的独立可执行文件路径。

导出为 ONNX

是仅保存检查点还是将模型导出为 ONNX 格式。

帧率

运行独立模式时使用的固定帧率,如果为 None,则不使用固定时间步长。

无头模式

标志,指示独立 Unreal Engine 进程是否应在无头模式下运行。

启动 Unreal

标志,指示脚本是否应启动独立的 Unreal Engine 进程。

日志设置

类型: LoggingSettings

map

启动独立的 Unreal Engine 进程时要加载的地图。

覆盖名称前缀

覆盖检查点文件的名称前缀(例如,SAC、PPO 等)。

网络架构设置

类型: NetworkArchitectureSettings

插件

类型: List[RLLibLauncherExtension]

port

连接到 Unreal Engine 进程的端口,如果为 None,则在独立模式运行时会自动选择一个开放的端口。

资源设置

类型: ResourceSettings

恢复设置

类型: ResumeSettings

保存最终策略

训练完成后是否保存最终策略。

保存频率

保存检查点的频率。

训练设置

类型: TrainingSettings

方法

__init__

__init__(enable_checkpoints=False, checkpoint_dir='./ckpt', save_freq=100000, name_prefix_override=None, export_onnx=False, save_final_policy=False, launch_unreal=False, port=None, executable_path=None, headless=False, map=None, fps=None, disable_script=False, training_settings=, algorithm_settings=, logging_settings=, resume_settings=, network_architecture_settings=, resource_settings=, plugins=)

返回类型: None

make_unreal_connection

make_unreal_connection()

基于脚本参数创建 Unreal Engine 连接。

© . This site is unofficial and not affiliated with AMD.