Struct FRLlibNetworkArchSettings
Struct FRLlibNetworkArchSettings
- Defined in File RLlibNetworkArchitectureSettings.h
Inheritance Relationships
Base Type
public FTrainingSettings(Struct FTrainingSettings)
struct FRLlibNetworkArchSettings : public FTrainingSettingsSource: Source/Schola/Training/Public/TrainingSettings/Ray/RLlibNetworkArchitectureSettings.h
Dependencies: FScriptArgBuilder
A struct to hold network architecture settings for an RLlib training script.
Public Functions
| Symbol | Details |
|---|---|
GenerateTrainingArgs | Generate the training arguments for the script using the ArgBuilder. |
~FRLlibNetworkArchSettings | — |
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/Ray/RLlibNetworkArchitectureSettings.h
Implementation: Source/Schola/Training/Private/TrainingSettings/Ray/RLlibNetworkArchitectureSettings.cpp
~FRLlibNetworkArchSettings
virtual ~FRLlibNetworkArchSettings()Attributes: virtual
Source: Source/Schola/Training/Public/TrainingSettings/Ray/RLlibNetworkArchitectureSettings.h
Implementation: Source/Schola/Training/Private/TrainingSettings/Ray/RLlibNetworkArchitectureSettings.cpp
Public Members
| Symbol | Details |
|---|---|
ActivationFunction | The activation function to use in the neural network. |
FCNetHiddens | The number and width of hidden layers in the neural network. |
MinibatchSize | The size of each minibatch. |
bUseAttention | Whether to use attention in the model. |
AttentionDims | The dimension of the attention layer. |
Var
ERLlibActivationFunctionEnum ActivationFunction = ERLlibActivationFunctionEnum::ReLUThe activation function to use in the neural network.
FCNetHiddens
TArray<int> FCNetHiddens = {512, 512}The number and width of hidden layers in the neural network.
MinibatchSize
int MinibatchSize = 256The size of each minibatch.
bUseAttention
bool bUseAttention = falseWhether to use attention in the model.
AttentionDims
int AttentionDims = 64The dimension of the attention layer.