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