torchrl.trainers.algorithms.configs.hooks.SelectKeysConfig

class torchrl.trainers.algorithms.configs.hooks.SelectKeysConfig(keys: list[str] = <factory>, _target_: str = 'torchrl.trainers.trainers.SelectKeys')

Configuration for the SelectKeys hook.

Examples

>>> from torchrl.trainers.algorithms.configs.hooks import SelectKeysConfig
>>> from hydra.utils import instantiate
>>> hook = instantiate(SelectKeysConfig(keys=["observation", "action"]))