关于DGL的bug Data type not recognized with bits 16
关于DGL的bug Data type not recognized with bits 16
dgl._ffi.base.DGLError: [21:21:28] /opt/dgl/src/array/cuda/sddmm.cu:112: Data type not recognized with bits 16
这个bug可能的原因是模型计算时候的数据出现了16位的。
解决办法:
找到可能出现16位的变量,改成32位。
如果喂给模型的tensor都是32位的,还是报这个错,有可能的原因是,torch的a
原创
2022-03-03 21:57:15 ·
878 阅读 ·
0 评论