TrainerHookBase#
- class torchrl.trainers.TrainerHookBase[source]#
An abstract hooking class for torchrl Trainer class.
- abstract register(trainer: Trainer, name: str)[source]#
Registers the hook in the trainer at a default location.
- Parameters:
trainer (Trainer) – the trainer where the hook must be registered.
name (str) – the name of the hook.
Note
To register the hook at another location than the default, use
register_op().