Skip to content

Sb3NetworkArchitectureSettings

Full path: schola.scripts.sb3.train.settings.Sb3NetworkArchitectureSettings

Network architecture settings for SB3 algorithms.

Sb3NetworkArchitectureSettings(policy_parameters = <factory>,
                               critic_parameters = <factory>,
                               activation = ActivationFunctionEnum.ReLU)

Parameters

  • policy_parameters (Annotated)

  • critic_parameters (Annotated)

  • activation (ActivationFunctionEnum)

Methods

init

__init__(policy_parameters = <factory>, critic_parameters = <factory>,
         activation = ActivationFunctionEnum.ReLU)

Parameters

  • policy_parameters (Annotated)

  • critic_parameters (Annotated)

  • activation (ActivationFunctionEnum)

Attributes

activation

activation

Activation function to use in the policy and critic networks. This determines the non-linear activation function applied to each layer of the neural networks. The choice of activation function can affect the performance of the model and may depend on the specific characteristics of the environment.

critic_parameters

critic_parameters

A list of layer widths representing the critic (value function) network architecture. This defines the number of neurons in each hidden layer of the critic network. For example, [64, 64] would create a critic network with two hidden layers, each containing 64 neurons. This is only applicable for algorithms that use a critic (e.g., SAC). If set to None, it will use the default architecture defined by the algorithm.

policy_parameters

policy_parameters

A list of layer widths representing the policy network architecture. This defines the number of neurons in each hidden layer of the policy network. For example, [64, 64] would create a policy network with two hidden layers, each containing 64 neurons. If set to None, it will use the default architecture defined by the algorithm.