[RuntimeError] Expected floating point type for target with class probabilities, got Long_runtimeerror: expected floating point type for tar

作者：AllinToyou | 2024-05-31 02:17:50

踩

runtimeerror: expected floating point type for target with class probabiliti

一些报错
这个报错类型翻译过来就是：有类别概率的目标的预期浮点类型，得 Long
大概意思就是你输入的标签值是浮点数，但实际上你所获得的是Long类型的值
原代码

Train_L = torch.tensor(expect_output,dtype=torch.int64)   # int64
label = Variable(Train_L)
1
2

解决方法

Train_L = torch.tensor(expect_output,dtype=torch.int64)   # int64
label = Variable(Train_L).float()
1
2

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/AllinToyou/article/detail/649859