本文共 424 字,大约阅读时间需要 1 分钟。
在运行损失函数时 loss = loss_func(output, label) 时报了两个错 :
1.
RuntimeError: Expected object of scalar type Long but got scalar type Float for argument #2 'target'
这是因为label必须是 LongTensor 类型, 之前是 :
label = t.tensor(float(image[1])) # image[1]为str类型
改为 :
label = t.tensor(float(image[1])).long()
2.
Assertion `cur_target >= 0 && cur_target < n_classes’ failed
这是因为我的数据集类别是从1开始,需要依次减一,所以要改为 :
label = t.tensor(float(image[1])-1).long()
转载地址:http://wncq.baihongyu.com/