Skip to content

FSB3NetworkArchSettings

A struct to hold network architecture settings for an SB3 training script.

struct FSB3NetworkArchSettings : public FTrainingSettings

Methods

GenerateTrainingArgs

virtual void GenerateTrainingArgs(FScriptArgBuilder &ArgBuilder) const

Generate the training arguments for the script using the ArgBuilder.

Parameters

  • ArgBuilder (FScriptArgBuilder)

FSB3NetworkArchSettings

virtual ~FSB3NetworkArchSettings()

Attributes

ActivationFunction

ESB3ActivationFunctionEnum ActivationFunction = ESB3ActivationFunctionEnum::ReLU

The activation function to use in the neural network.


CriticParameters

TArray<int> CriticParameters = {256, 256}

The number and width of hidden layers in the Critic.


PolicyParameters

TArray<int> PolicyParameters = {256, 256}

The number and width of hiddent layers in the policy network.

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