FRLlibCheckpointSettings

A struct to hold checkpoint settings for an RLlib training script.

struct FRLlibCheckpointSettings : public FTrainingSettings

Methods

GenerateTrainingArgs

virtual void GenerateTrainingArgs(FScriptArgBuilder &ArgBuilder) const

Generate the training arguments for the script using the ArgBuilder.

Parameters

• ArgBuilder (FScriptArgBuilder)

---

FRLlibCheckpointSettings

virtual ~FRLlibCheckpointSettings()

Attributes

bSaveFinalModel

bool bSaveFinalModel = true

Whether to save the final output of training.

---

bExportToONNX

bool bExportToONNX = false

Whether to export the final model to ONNX automatically.

---

bEnableCheckpoints

bool bEnableCheckpoints = false

Whether to save checkpoints during training.

---

SaveFreq

int SaveFreq = 1000

How frequently should we make checkpoints.

---

CheckpointDir

FDirectoryPath CheckpointDir

The directory to save checkpoints to.

Source: Source/ScholaTraining/Public/TrainingSettings/Ray/RLlibCheckpointSettings.h