在训练的train.py改为:
running_corrects += float(torch.sum(preds == labels.data))
是不是pytorch的版本问题我不知道,主要原因还是因为sum中的数值如果是bool类型,那么输出的就只有0和1,所以要将sum中数据类型强行转换,转换成float的类型,就会解决这种情况
在训练的train.py改为:
running_corrects += float(torch.sum(preds == labels.data))
是不是pytorch的版本问题我不知道,主要原因还是因为sum中的数值如果是bool类型,那么输出的就只有0和1,所以要将sum中数据类型强行转换,转换成float的类型,就会解决这种情况