跳至内容

FRLlibCheckpointSettings

struct FRLlibCheckpointSettings : public FTrainingSettings

一个用于保存 RLlib 训练脚本的检查点设置的结构体。

依赖项: FScriptArgBuilder, FTrainingSettings

继承自: public FTrainingSettings

公共接口

析构函数

~FRLlibCheckpointSettings

virtual ~FRLlibCheckpointSettings()

属性: virtual

Source: Schola/Source/Schola/Public/Subsystem/SubsystemSettings/Ray/RLlibCheckpointSettings.h (line 40, column 9)

Implementation: Schola/Source/Schola/Private/Subsystem/SubsystemSettings/Ray/RLlibCheckpointSettings.cpp (lines 24-27)

公共函数

GenerateTrainingArgs

virtual void GenerateTrainingArgs(int Port, FScriptArgBuilder &ArgBuilder) const const

使用 ArgBuilder 为脚本生成训练参数。

注意:端口作为脚本的常见参数传入,并在高层设置,但特定子设置可能需要它。

参数

  • Port (int) – [输入] 用于脚本的端口
  • ArgBuilder (FScriptArgBuilder &) – [输入] 用于生成参数的构建器

属性: const, virtual

Source: Schola/Source/Schola/Public/Subsystem/SubsystemSettings/Ray/RLlibCheckpointSettings.h (line 38, column 6)

Implementation: Schola/Source/Schola/Private/Subsystem/SubsystemSettings/Ray/RLlibCheckpointSettings.cpp (lines 6-22)

公共成员

bool bSaveFinalModel

bool bSaveFinalModel = = true

是否保存训练的最终输出。

Source: Schola/Source/Schola/Public/Subsystem/SubsystemSettings/Ray/RLlibCheckpointSettings.h (line 20, column 6)

bool bExportToONNX

bool bExportToONNX = = false

是否自动将最终模型导出为 ONNX。

Source: Schola/Source/Schola/Public/Subsystem/SubsystemSettings/Ray/RLlibCheckpointSettings.h (line 24, column 6)

bool bEnableCheckpoints

bool bEnableCheckpoints = = false

训练期间是否保存检查点。

Source: Schola/Source/Schola/Public/Subsystem/SubsystemSettings/Ray/RLlibCheckpointSettings.h (line 28, column 6)

int SaveFreq

int SaveFreq = = 1000

应多久保存一次检查点。

Source: Schola/Source/Schola/Public/Subsystem/SubsystemSettings/Ray/RLlibCheckpointSettings.h (line 32, column 5)

FDirectoryPath CheckpointDir

FDirectoryPath CheckpointDir

保存检查点的目录。

Source: Schola/Source/Schola/Public/Subsystem/SubsystemSettings/Ray/RLlibCheckpointSettings.h (line 36, column 16)

被使用: FRLlibTrainingSettings

Source: Schola/Source/Schola/Public/Subsystem/SubsystemSettings/Ray/RLlibCheckpointSettings.h (line 13, column 1)

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