错误代码
RuntimeError: mixed dtype (CPU): expect input to have scalar type of BFloat16
产生原因
- 将 tensor 与 numpy array 做运算，并将 Layernorm 应用到其计算结果。
解决方法
- 将 numpy array 换为 tensor。
RuntimeError: mixed dtype (CPU): expect input to have scalar type of BFloat16
2022-10-03
近期评论