快捷方式

ConstantKLController

class torchrl.data.ConstantKLController(*, kl_coef: float | None = None, model: nn.Module | None = None)[source]

Constant KL Controller。

此控制器在收到更新值时保持固定的系数,无论其值如何。

关键字参数:
  • kl_coeffloat)– 在计算奖励时用于乘以 KL 的系数。

  • modelnn.Module, optional)– 需要控制的包装模型。必须具有 "kl_coef" 属性。如果提供,则 "kl_coef" 将就地更新。

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

为初学者和高级开发者提供深入的教程

查看教程

资源

查找开发资源并让您的问题得到解答

查看资源