为什么cifar10 训练过程中使用_, pred = torch.max(outputs.data, dim=1)

来源：6-6 PyTorch搭建 VGGNet 实现Cifar10图像分类

慕莱坞8405259

2025-03-23

_, pred = torch.max(outputs.data, =)

correct = pred.eq(labels.data).cpu().sum()

这里为什么使用

labels.data
outputs.data

直接使用

_, pred = torch.max(outputs, =)

correct = pred.eq(labels).cpu().sum()

写回答

1回答

会写代码的好厨师

2025-03-24

两个用法都可以。

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

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

1190 学习 · 298 问题

相似问题

回答 1

回答 1

回答 1

回答 1

回答 1