测试准确率计算方法说明 pre.eq(target).float().sum().item()

测试准确率计算方法说明 pre.eq(target).float().sum().item()

待办

pred = logits.argmax(dim=1)
    correct += pred.eq(target).float().sum().item()
	
	这句意思就是输出最大值的索引位置,这个索引位置和真实值的索引位置比较相等的做统计就是这个批次准确的个数用来做统计分析

猜你喜欢

转载自www.cnblogs.com/lishikai/p/12367305.html
今日推荐