Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

关于internvl2.5中第一阶段训练mlp的数据格式 #787

Open
SuHuaFeng opened this issue Dec 23, 2024 · 2 comments
Open

关于internvl2.5中第一阶段训练mlp的数据格式 #787

SuHuaFeng opened this issue Dec 23, 2024 · 2 comments

Comments

@SuHuaFeng
Copy link

您好,我想请教一下第一阶段训练mlp的数据格式是什么样的呢,我需要怎么构建数据集去做第一阶段的训练?

@Weiyun1025
Copy link
Collaborator

可以参考我们的文档

@SuHuaFeng
Copy link
Author

我现在把llm换成了qwen2-coder-7b,这是一个代码生成模型,然后我进行了第一阶段的训练,我使用10000步的checkpoint推理发现结果很差(如下图),
image
我觉得可能是训练的步数不够也可能是我的训练数据格式就是错的,下图是我目前的训练数据格式,输入是图像和指令,gpt输出是代码,
image
看了论文,第一阶段做的是对比学习,输入是不是应该是图像和代码,然后gpt的输出应该是一个标签,0或者1,代表图像和代码是否匹配,然后之后进行第二阶段的全参微调才使用我目前这样的格式。
希望您能解答一下我的疑问,应该怎么修改数据格式,然后我的替换llm的这个想法在实现的过程需要注意些什么,万分感激!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants