在用深度学习模型跑文本分类任务时,运行模型时报这个错误了,检查发现传入模型的embedding_matrix的类型不是float,而是double,所以要做类型转换
‘’’
embedding_matrix = torch.from_numpy(embedding_matrix).float()
‘’’
Expected object of type torch.cuda.DoubleTensor but found type torch.cuda.FloatTensor
最新推荐文章于 2022-03-21 09:52:50 发布