Skip to content

Struct FRLlibNetworkArchSettings

Struct FRLlibNetworkArchSettings

  • Defined in File RLlibNetworkArchitectureSettings.h

Inheritance Relationships

Base Type

struct FRLlibNetworkArchSettings : public FTrainingSettings

Source: Source/Schola/Training/Public/TrainingSettings/Ray/RLlibNetworkArchitectureSettings.h

Dependencies: FScriptArgBuilder

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


Public Functions

SymbolDetails
GenerateTrainingArgsGenerate the training arguments for the script using the ArgBuilder.
~FRLlibNetworkArchSettings

GenerateTrainingArgs

virtual void GenerateTrainingArgs(FScriptArgBuilder &ArgBuilder) const

Generate 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.

#DirectionNameTypeDescription
1ArgBuilderFScriptArgBuilder &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

SymbolDetails
ActivationFunctionThe activation function to use in the neural network.
FCNetHiddensThe number and width of hidden layers in the neural network.
MinibatchSizeThe size of each minibatch.
bUseAttentionWhether to use attention in the model.
AttentionDimsThe dimension of the attention layer.

Var

ERLlibActivationFunctionEnum ActivationFunction =
ERLlibActivationFunctionEnum::ReLU

The 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 = 256

The size of each minibatch.


bUseAttention

bool bUseAttention = false

Whether to use attention in the model.


AttentionDims

int AttentionDims = 64

The dimension of the attention layer.