TrainerHookBase

class torchrl.trainers.TrainerHookBase

An abstract hooking class for torchrl Trainer class.

abstract register(trainer: Trainer, name: str)

Registers the hook in the trainer at a default location.

参数:

• trainer (Trainer) – the trainer where the hook must be registered.

• name (str) – the name of the hook.

注意

To register the hook at another location than the default, use register_op().