OptimizerHook

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

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

引數:

• optimizer (optim.Optimizer) – 用於應用於損失元件的最佳化器。

• loss_components (Sequence[str], 可選) – Loss 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()。