为什么验证时loss要除以64而accuracy不用?

来源:4-7 利用神经网络解决分类和回归问题(5)

谢思阳

2020-12-06

accuracy = accuracy / len(test_data)
loss_test = loss_test / (len(test_data) // 64)

demo_cls.py line 65-66

写回答

1回答

woryo

2020-12-24

因为代码里算的loss是一次iteration下的平均损失,loss_test先在一个iteration内累加,累加计算完成后自然需要除以(len(test_data)//64)来取均值; accuracy准确率计算公式就是推理正确的样本数量/测试集样本总数。

1
0

PyTorch入门到进阶 实战计算机视觉与自然语言处理项目

理论基础+技术讲解+实战开发,快速掌握PyTorch框架

1190 学习 · 293 问题

查看课程