Skip to content

为什么ppo model 需要接AutoModelForCausalLMWithValueHead呢? #51

Description

@jiahuanluo

感谢工作!
请问这里 ppo model 为什么要接一个valuehead 呢?
https://github.com/beyondguo/LLM-Tuning/blob/ed68123815bc0add9ad2d7ddc2a48dc584db2c94/RLHF/rl_training.py#L185C1-L185C11
这个head好像随机初始化的?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions