pytorch深度学习原理实战-rightness函数
在pytorch深度学习实战(集智俱乐部)书中卷积神经网络一章没有定义rightness函数。下面是自己实现代码def rightness(output, target): preds = output.data.max(dim=1, keepdim=True)[1] return preds.eq(target.data.view_as(preds)).cpu().sum(),...
原创
2020-01-04 15:42:04 ·
2007 阅读 ·
4 评论