RuntimeError: mixed dtype (CPU): expect input to have scalar type of BFloat16

错误代码

  • RuntimeError: mixed dtype (CPU): expect input to have scalar type of BFloat16

产生原因

  • 将 tensor 与 numpy array 做运算,并将 Layernorm 应用到其计算结果。

解决方法

  • 将 numpy array 换为 tensor。

版权声明:
作者:MWHLS
链接:https://mwhls.top/4011.html
来源:无镣之涯
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
打赏
< <上一篇
下一篇>>