1
1
数学库
deepspeed版本0.12.3+gita724046.abi1.dtk2404.torch2.1.0,代码有bug。
2024-12-10 20:19:46
99

attachmentId-image.png

grad_norms.append(g.to(get_accelerator().device_name(), non_blocking=True).float.norm(2))
这里的float后面是不是缺了一个()导致代码执行报错

全部评论(2)
只看楼主
C
changhl
2楼
您好:内部核对后确实存在该bug,目前有两个解决方案:1.将0.12.3的deepspeed处的float修改为float();2.参考您的这个问题的回复:https://forum.sourcefind.cn/thread/740,使用dtk24.04.2后的dtk,deepspeed版本对应0.14.2的版本(推荐)
2024/12/11
C
changhl
1楼

可以麻烦您提供一下您这边执行有bug的demo程序么?我们内部核对一下;

如果终端输出的bug调用栈明确指向这里,可以试下按您这边的结论修改一下进行对比呢。
2024/12/11
没有更多内容了
--
楼主
Level.1
0 帖子
楼主热帖
暂无数据