修改为:X[:, self.feature_index[feat][0]:self.feature_index[feat][1]].cpu().long().to(device))同时 predict函数中的y_pred要.cpu()在做后边的计算
这只是最简单的改法,最好是在进入mlu之前就把这个tensor to到cpu上换成long型
X[:, self.feature_index[feat][0]:self.feature_index[feat][1]].cpu().long().to(device))
同时
predict函数中的y_pred要.cpu()在做后边的计算
您好,nn.Embedding这个算子不支持float类型传入,我们mlu也是遵循这个原则,因此需要在cpu上将X[:, self.feature_index[feat][0]:self.feature_index[feat][1]]
这个转为Long型在to到mlu上即可避开这个问题。
尊敬的开发者您好,我们会安排人力确认下。
请登录后评论