问题描述
在运行别人代码时出现如下报错:
RuntimeError: expected scalar type Long but found Float
报错代码为:
data = list_text[1: 11]
label = list_text[0]
# print(data)
# print(label)
##转tensor
data = torch.tensor(data) #这行报错
label = torch.tensor(label)
print(data.shape)
print(label.shape)
原因分析:
提示的错误是关于类型的问题,把列表变张量时,torch.long需要变torch.float类型
解决方案:
在如下代码位置加上,dtype显示地设置成torch.float
data = torch.tensor(data, dtype=torch.float) #这行报错
label = torch.tensor(label, dtype=torch.float)
报错解决!