© 1999-2048 dssz.net 粤ICP备11031372号
[其它] Pytorch 实现计算分类器准确率(总分类及子分类)
说明:分类器平均准确率计算: correct = torch.zeros(1).squeeze().cuda() total = torch.zeros(1).squeeze().cuda() for i, (images, labels) in enumerate(train_loader): images = Variable(images.cuda()) labels = Variable(labels.cuda()) output = model(imag<weixin_38685538> 上传 | 大小:27kb