TrainerHookBase

class torchrl.trainers.TrainerHookBase

torchrl Trainer 類別的抽象掛鉤類別。

abstract register(trainer: Trainer, name: str)

在預設位置將掛鉤註冊到訓練器中。

參數:

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

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

注意

若要將掛鉤註冊到預設位置以外的其他位置，請使用 register_op()。