快捷方式

ConstantKLController

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

恒定 KL 控制器。

此控制器无论使用什么值进行更新,都保持固定系数。

关键字参数:
  • kl_coef (float) – 计算奖励时乘以 KL 的系数。

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源