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

Question

_, 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()

Answer 1

会写代码的好厨师 2025-03-24 10:53:50

 两个用法都可以。