Accuracy一直显示0

在训练的train.py改为:

running_corrects += float(torch.sum(preds == labels.data))

是不是pytorch的版本问题我不知道,主要原因还是因为sum中的数值如果是bool类型,那么输出的就只有0和1,所以要将sum中数据类型强行转换,转换成float的类型,就会解决这种情况

猜你喜欢

转载自blog.csdn.net/weixin_36411839/article/details/103497868
今日推荐