torchrl.trainers.algorithms.configs.modules.TanhNormalModelConfig

class torchrl.trainers.algorithms.configs.modules.TanhNormalModelConfig(_partial_: bool = False, in_keys: Optional[Any] = None, out_keys: Optional[Any] = None, network: MLPConfig = '???', eval_mode: bool = False, extract_normal_params: bool = True, scale_mapping: str = 'biased_softplus_1.0', scale_lb: float = 0.0001, param_keys: Optional[Any] = None, exploration_type: Any = 'RANDOM', return_log_prob: bool = False, _target_: str = 'torchrl.trainers.algorithms.configs.modules._make_tanh_normal_model')

A class to configure a TanhNormal model.

Example

>>> cfg = TanhNormalModelConfig(network=MLPConfig(in_features=10, out_features=5, depth=2, num_cells=32))
>>> net = instantiate(cfg)
>>> y = net(torch.randn(1, 10))
>>> assert y.shape == (1, 5)

See also

torchrl.modules.TanhNormal