TrainerHookBase

class torchrl.trainers.TrainerHookBase

Torchrl Trainer 类的抽象挂钩类。

abstract register(trainer: Trainer, name: str)

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

参数:

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

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

注意

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