reagent.net_builder.value package

Submodules

reagent.net_builder.value.fully_connected module

class reagent.net_builder.value.fully_connected.FullyConnected(sizes: List[int] = <factory>, activations: List[str] = <factory>, use_layer_norm: bool = False)

Bases: reagent.net_builder.value_net_builder.ValueNetBuilder

activations: List[str]

build_value_network(state_normalization_data: reagent.core.parameters.NormalizationData, output_dim: int = 1) → torch.nn.modules.module.Module

sizes: List[int]

use_layer_norm: bool = False

reagent.net_builder.value.seq2reward_rnn module

class reagent.net_builder.value.seq2reward_rnn.Seq2RewardNetBuilder(action_dim: int = 2, num_hiddens: int = 64, num_hidden_layers: int = 2)

Bases: reagent.net_builder.value_net_builder.ValueNetBuilder

action_dim: int = 2

build_value_network(state_normalization_data: reagent.core.parameters.NormalizationData) → torch.nn.modules.module.Module

num_hidden_layers: int = 2

num_hiddens: int = 64

Module contents