torchrl.trainers.algorithms.configs.objectives.IQLLossConfig¶
- class torchrl.trainers.algorithms.configs.objectives.IQLLossConfig(_partial_: bool = False, actor_network: Any = None, qvalue_network: Any = None, value_network: Any = None, discrete: bool = False, action_space: Any = None, num_qvalue_nets: int = 2, loss_function: str = 'smooth_l1', temperature: float = 1.0, expectile: float = 0.5, gamma: float | None = None, priority_key: str | None = None, separate_losses: bool = False, reduction: str | None = None, deactivate_vmap: bool = False, scalar_output_mode: str | None = None, _target_: str = 'torchrl.trainers.algorithms.configs.objectives._make_iql_loss')[source]¶
Hydra configuration for
IQLLoss(andDiscreteIQLLosswhendiscrete=True).Every kwarg accepted by
IQLLoss.__init__is exposed as a field here. Thediscrete/action_spacefields apply only when the discrete variant is selected.