TrainerHookBase

class torchrl.trainers.TrainerHookBase

torchrl Trainer 类的抽象 Hooking 类。

abstract register(trainer: Trainer, name: str)

在训练器的默认位置注册 hook。

参数:

• trainer (Trainer) – 必须注册 hook 的训练器。

• name (str) – hook 的名称。

注意

要在默认位置之外的其他位置注册 hook，请使用 register_op()。