Skip to content

FSB3TrainingSettings

Configuration struct for Stable Baselines 3 (SB3) training scripts.

Contains all settings necessary to launch and configure an SB3 training session. This is a partial implementation of the available SB3 settings and is not exhaustive.

struct FSB3TrainingSettings : public FTrainingSettings

Methods

GenerateTrainingArgs

virtual void GenerateTrainingArgs(FScriptArgBuilder &ArgBuilder) const

Generate command-line arguments for the SB3 training script.

Parameters

  • ArgBuilder (FScriptArgBuilder)

FSB3TrainingSettings

virtual ~FSB3TrainingSettings()

Virtual destructor.

Attributes

Timesteps

int Timesteps = 8000

The total number of environment timesteps to train for.


LoggingSettings

FSB3LoggingSettings LoggingSettings

Settings for logging training metrics and outputs.


CheckpointSettings

FSB3CheckpointSettings CheckpointSettings

Settings for model checkpointing during training.


ResumeSettings

FSB3ResumeSettings ResumeSettings

Settings for resuming training from a checkpoint.


NetworkArchitectureSettings

FSB3NetworkArchSettings NetworkArchitectureSettings

Settings for neural network architecture (e.g.


bDisplayProgressBar

bool bDisplayProgressBar = true

Whether to display a progress bar during training.


bDisableEval

bool bDisableEval = false

When true, skips post-training evaluation (passes —disable-eval to the script).


Algorithm

ESB3TrainingAlgorithm Algorithm = ESB3TrainingAlgorithm::PPO

The reinforcement learning algorithm to use (e.g., SAC, PPO).


PPOSettings

FSB3PPOSettings PPOSettings

PPO specific settings.


SACSettings

FSB3SACSettings SACSettings

SAC specific settings.

Source: Source/ScholaTraining/Public/TrainingSettings/StableBaselines/SB3TrainingSettings.h