测试准确率计算方法说明 pre.eq(target).float().sum().item()
待办
pred = logits.argmax(dim=1)
correct += pred.eq(target).float().sum().item()
这句意思就是输出最大值的索引位置,这个索引位置和真实值的索引位置比较相等的做统计就是这个批次准确的个数用来做统计分析
待办
pred = logits.argmax(dim=1)
correct += pred.eq(target).float().sum().item()
这句意思就是输出最大值的索引位置,这个索引位置和真实值的索引位置比较相等的做统计就是这个批次准确的个数用来做统计分析
本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。