TrainerHookBase

class torchrl.trainers.TrainerHookBase

torchrl Trainer 類的一個抽象鉤子類。

abstract register(trainer: Trainer, name: str)

在訓練器中的預設位置註冊鉤子。

引數:

• trainer (Trainer) – 鉤子必須註冊到的訓練器。

• name (str) – 鉤子的名稱。

注意

要在預設位置以外的位置註冊鉤子，請使用 register_op()。