Struct FRLlibCheckpointSettings

struct FRLlibCheckpointSettings : public FTrainingSettings

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

Public Functions

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

Generate the training arguments for the script using the ArgBuilder.

Note

port is supplied as it is a common argument to pass to scripts, and is set at a high level but might be needed by specific subsettings

Parameters:

• Port – [in] The port to use for the script

• ArgBuilder – [in] The builder to use to generate the arguments

virtual ~FRLlibCheckpointSettings()

Public Members

bool bSaveFinalModel = true

Whether to save the final output of training.

bool bExportToONNX = false

Whether to export the final model to ONNX automatically.

bool bEnableCheckpoints = false

Whether to save checkpoints during training.

int SaveFreq = 1000

How frequently should we make checkpoints.

FDirectoryPath CheckpointDir

The directory to save checkpoints to.