Merge branch 'main' into master

Author: woongjoonchoi

beginner_source/blitz/cifar10_tutorial.py

@@ -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()
 

prototype_source/flight_recorder_tutorial.rst

@@ -214,7 +214,6 @@ For demonstration purposes, we named this program ``crash.py``.
    complexities.
 
 .. code:: python
-  :caption: A crashing example
 
   import torch
   import torch.distributed as dist