Struct FSB3CheckpointSettings

struct FSB3CheckpointSettings : public FTrainingSettings

A struct to hold checkpoint settings for an SB3 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 ~FSB3CheckpointSettings()

Public Members

bool bSaveCheckpoints = false

Whether to save checkpoints during training.

bool bSaveFinalModel = true

Whether to save the final model.

bool bSaveVecNormalize = false

Whether to save the VecNormalize parameters.

bool bSaveReplayBuffer = false

Whether to save the replay buffer.

bool bExportFinalModelToOnnx = true

Whether we should just create a checkpoint, or if we should also export the final model out to ONNX.

FDirectoryPath CheckpointDir

The directory to save checkpoints to.

int SaveFreq = 1000

The frequency to save checkpoints.

FString NamePrefix = FString(“ppo”)

The prefix to use for the checkpoint files.