OptimizerHook

class torchrl.trainers.OptimizerHook(optimizer: Optimizer, loss_components: Optional[Sequence[str]] = None)

為一個或多個損失組件新增最佳化器。

參數:

• optimizer (optim.Optimizer) – 要應用於 loss_components 的最佳化器。

• loss_components (Sequence[str], optional) – 損失 TensorDict 中最佳化器應應用於相應值的鍵。如果省略，最佳化器將應用於名稱以 loss_ 開頭的所有組件。

範例

>>> optimizer_hook = OptimizerHook(optimizer, ["loss_actor"])
>>> trainer.register_op("optimizer", optimizer_hook)

register(trainer, name='optimizer') → None

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

參數:

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

• name (str) – hook 的名稱。

注意

要在預設以外的其他位置註冊 hook，請使用 register_op()。