Skip to content

Commit

Permalink
Remove Tensor.data attribute from tutorial (#3193)
Browse files Browse the repository at this point in the history
  • Loading branch information
HanBnrd authored Dec 20, 2024
1 parent 7262c21 commit 2793639
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion beginner_source/blitz/cifar10_tutorial.py
Original file line number Diff line number Diff line change
Expand Up @@ -252,7 +252,7 @@ def forward(self, x):
# calculate outputs by running images through the network
outputs = net(images)
# the class with the highest energy is what we choose as prediction
_, predicted = torch.max(outputs.data, 1)
_, predicted = torch.max(outputs, 1)
total += labels.size(0)
correct += (predicted == labels).sum().item()

Expand Down

0 comments on commit 2793639

Please sign in to comment.