RuntimeError: expected scalar type Long but found Float 报错解决

问题描述

在运行别人代码时出现如下报错:

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)

报错解决!

你可能感兴趣的:(深度学习,python,机器学习,pytorch,经验分享)