Struct FSB3NetworkArchSettings
Struct FSB3NetworkArchSettings
- Defined in File SB3NetworkArchitectureSettings.h
Inheritance Relationships
Base Type
public FTrainingSettings(Struct FTrainingSettings)
struct FSB3NetworkArchSettings : public FTrainingSettingsSource: Source/Schola/Training/Public/TrainingSettings/StableBaselines/SB3NetworkArchitectureSettings.h
Dependencies: FScriptArgBuilder
A struct to hold network architecture settings for an SB3 training script.
Public Functions
| Symbol | Details |
|---|---|
GenerateTrainingArgs | Generate the training arguments for the script using the ArgBuilder. |
~FSB3NetworkArchSettings | — |
GenerateTrainingArgs
virtual void GenerateTrainingArgs(FScriptArgBuilder &ArgBuilder) constGenerate the training arguments for the script using the ArgBuilder.
Populates the ArgBuilder with training-specific command-line arguments.
Parameters
ArgBuilder – [inout] The builder to use to generate the arguments.
| # | Direction | Name | Type | Description |
|---|---|---|---|---|
| 1 | — | ArgBuilder | FScriptArgBuilder & | The builder to use to generate the arguments. |
Attributes: const, virtual
Source: Source/Schola/Training/Public/TrainingSettings/StableBaselines/SB3NetworkArchitectureSettings.h
Implementation: Source/Schola/Training/Private/TrainingSettings/StableBaselines/SB3NetworkArchitectureSettings.cpp
~FSB3NetworkArchSettings
virtual ~FSB3NetworkArchSettings()Attributes: virtual
Source: Source/Schola/Training/Public/TrainingSettings/StableBaselines/SB3NetworkArchitectureSettings.h
Implementation: Source/Schola/Training/Private/TrainingSettings/StableBaselines/SB3NetworkArchitectureSettings.cpp
Public Members
| Symbol | Details |
|---|---|
ActivationFunction | The activation function to use in the neural network. |
CriticParameters | The number and width of hidden layers in the Critic. |
PolicyParameters | The number and width of hiddent layers in the policy network. |
ActivationFunction
ESB3ActivationFunctionEnum ActivationFunction = ESB3ActivationFunctionEnum::ReLUThe activation function to use in the neural network.
CriticParameters
TArray<int> CriticParameters = {256, 256}The number and width of hidden layers in the Critic.
Applied to either the Q-Function or Value-Function
PolicyParameters
TArray<int> PolicyParameters = {256, 256}The number and width of hiddent layers in the policy network.