reagent.net_builder.parametric_dqn package

Submodules

reagent.net_builder.parametric_dqn.fully_connected module

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

Bases: reagent.net_builder.parametric_dqn_net_builder.ParametricDQNNetBuilder

activations: List[str]

build_q_network(state_normalization_data: reagent.core.parameters.NormalizationData, action_normalization_data: reagent.core.parameters.NormalizationData, output_dim: int = 1) → reagent.models.base.ModelBase

sizes: List[int]

use_batch_norm: bool = False

use_layer_norm: bool = False

Module contents