已合并
新增优化器权重转换nooplayer/dpp功能 #2190
温一盏创建于 2025年2月7日
新增优化器权重转换nooplayer/dpp功能 #2190
已合并
温一盏创建于 2025年2月7日
refs/pull/2190/head合入到master
温一盏
温一盏成员
2025年2月7日

新增优化器权重转换nooplayer/dpp功能,验证如下:
https://wiki.huawei.com/domains/64729/wiki/101436/WIKI202502075883566

1.增加优化器权重转换的动态流水线划分功能;
2.增加优化器权重转换的空层功能;
3.支持带有空层的mcore-mcore权重转换,并新增参数'--load-checkpoint-loosely',仅用于带有空层的mcore-mcore权重转换;
4.修改优化器脚本,删除多余参数;
5.修改权重转换readme

likedislike
Pull Request已成功合入, 合并人@
(感谢 温一盏 的贡献)
温一盏温一盏成员
2025年2月7日 指派了 fengliangjun 参与评审
温一盏温一盏成员
2025年2月7日 指派了 王姜奔 参与评审
AAtomGit-Bot
2025年2月7日 添加了   ascend-cla/yes 标签
温一盏温一盏成员
2025年2月7日 更新了说明
温一盏
温一盏成员
2025年2月7日 评论:

compile

likedislike
此处折叠了42条消息 查看更多
AtlasAccountAtlasAccount成员
2025年2月10日 移除了   ci-pipeline-failed 标签
温一盏
温一盏成员2025年2月17日进行代码检视1
mindspeed_llm/tasks/checkpoint/loader_mg.py
@@ -56,3 +56,3 @@
5656 group.add_argument("--add-output-layer-bias", action="store_true", default=False,
5757 help='Configuration for the output layer bias.')
58- 
58+ group.add_argument('--load-checkpoint-loosely', action='store_true', default=False,
温一盏
温一盏2025年2月17日评论:

在带有空层的mg-mg时,空层的权重key缺失,load报错,需要松弛加载

likedislike
温一盏
温一盏成员2025年2月17日进行代码检视1
mindspeed_llm/tasks/checkpoint/optim_converter.py
@@ -547,2 +546,2 @@
547- self.set_optimizer_layer_norm(src_data, dst_dict, i)
548- self.set_optimizer_layer_mlp(src_data, dst_dict, i)
546+ flag = False
547+ if len(src_data[i].keys()) > 0:
温一盏
温一盏2025年2月17日评论:

此处为判断模型第一层与最后一层之间的中间层是否为空层

likedislike
温一盏
温一盏成员
2025年2月17日 评论:

compile

likedislike
AAtomGit-Bot
2025年2月18日 添加了   approved 标签