TrainerHookBase

class torchrl.trainers.TrainerHookBase

torchrl Trainer 类的一个抽象钩子类。

abstract register(trainer: Trainer, name: str)

在训练器中的默认位置注册钩子。

参数:

• trainer (Trainer) – 钩子必须注册到的训练器。

• name (str) – 钩子的名称。

注意

要在默认位置以外的位置注册钩子，请使用 register_op()。