已关闭
[Bug]: 使用mindspeed-mm 2.30和主干版本进行推理微调,发现从dcp转成hf权重参数后增长一倍的问题 #250
wilxy创建于  2月12日关闭于  4月23日
wilxy
wilxy
2月12日 创建

在提交新问题之前,请确保您已经在社区中搜索过相关问题,并使用了社区中提供的资源/工具后,仍未找到满意的解决方式。

⚠️ 安全信息提醒:请仔细检查提供的文本内容,确保其不包含敏感数据信息,包括但不限于:

  • API 令牌或密钥
  • 密码或身份验证凭证
  • 私有网址或接口地址
  • 个人或机密数据
  • ...

在分享配置信息或代码示例时,请将敏感信息脱敏处理,或使用 <TOKEN> 等占位符替代原有内容。

环境信息

mindspeed-mm:main和2.3.0

🐛 问题描述

【问题描述】
原生的hf权重:
image.png
默认mindspeed-main 微调出dcp重新转化成hf权重:
image.png
【问题分析】
发现默认微调后的模型从float16变成float32导致:
原有模型dtype:
image.png
 默认微调后的dtype:
image.png
【问题跟踪】且修改fsdp2_config.yaml中的参数无效
image.png
image.png

欢迎加入社区,感谢您对社区的贡献 🎉!

likedislike
ascend-robotascend-robot成员
2月12日 添加了label:bug
ningmenglhningmenglh成员
2月13日 将 ningmenglh 设为负责人
ningmenglh
ningmenglh成员
2月13日 评论:

感谢您使用MindSpeed-MM,这个问题解释如下:
原始huggingface权重数据类型为bf16,所以权重大小为17G,在经过训练之后,保存的权重数据类型为高精的fp32,其大小刚好为原始权重的两倍
后续MM仓会加上相关说明,并添加保存为bf16数据类型的可选项

likedislike
wilxy
wilxy
2月14日 评论:

感谢您使用MindSpeed-MM,这个问题解释如下:
原始huggingface权重数据类型为bf16,所以权重大小为17G,在经过训练之后,保存的权重数据类型为高精的fp32,其大小刚好为原始权重的两倍
后续MM仓会加上相关说明,并添加保存为bf16数据类型的可选项

@ningmenglh

我在2.3.0 分支也增加了参数适配https://gitcode.com/Ascend/MindSpeed-MM/pull/2182

likedislike
ningmenglh
ningmenglh成员
2月26日 评论:

感谢您使用MindSpeed-MM,这个问题解释如下:
原始huggingface权重数据类型为bf16,所以权重大小为17G,在经过训练之后,保存的权重数据类型为高精的fp32,其大小刚好为原始权重的两倍
后续MM仓会加上相关说明,并添加保存为bf16数据类型的可选项

@ningmenglh

我在2.3.0 分支也增加了参数适配https://gitcode.com/Ascend/MindSpeed-MM/pull/2182

@wilyw

感谢为MindSpeed-MM提供相应的修改PR,在下一次评审会考虑合入该PR,不合入也会提供相应的替代方案

likedislike
Jjs1234567成员
3月10日 关联了pull request:[Modify]support dcp_to_hf can select float16
ningmenglh
ningmenglh成员
3月27日 评论:

已经在主线合入

likedislike
ningmenglhningmenglh成员
3月27日 issue状态由 TODO 改变为 DONE
ningmenglhningmenglh成员
3月27日 关闭了 issue
wilxy
wilxy
3月30日 评论:

2.3 版本麻烦也合下,代码稍微不同,代码也已review。
ttps://gitcode.com/Ascend/MindSpeed-MM/merge_requests/2182

likedislike
wilxywilxy
3月31日 issue状态由 DONE 改变为 Analysing
wilxywilxy
3月31日 重新打开了 issue
wilxywilxy
3月31日 issue状态由 Analysing 改变为 ACCEPTED
ningmenglh
ningmenglh成员
4月23日 评论:

感谢使用MindSpeed_MM,目前MindSpeed已经发布最新版本26.0.0以及最新分支master,都已经适配保存bf16类型权重的功能,请优先使用这两个版本的MindSpeed_MM仓

likedislike
ningmenglh
ningmenglh成员
4月23日 评论:

https://gitcode.com/Ascend/MindSpeed-MM/pull/2182 这个pr如检视意见一般,如果是非moe模型,if num_experts逻辑无法进入,则有错误风险,可选择使用最新分支以及26.0.0分支

likedislike
ningmenglhningmenglh成员
4月23日 issue状态由 ACCEPTED 改变为 DONE
ningmenglhningmenglh成员
4月23日 关闭了 issue