llama2_reward_7b¶

torchtune.models.llama2.llama2_reward_7b() → TransformerDecoder[source]¶

用于创建Llama2模型的构建器，使用默认的7B参数值进行初始化，来自https://arxiv.org/abs/2307.09288，其中输出层是一个分类层，投影到单个类别以进行奖励建模。