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()。