Merge branch 'main' into atskae/reduce_iterations_usb

Author: atskae

beginner_source/introyt/autogradyt_tutorial.py

@@ -250,9 +250,9 @@ class TinyModel(torch.nn.Module):
     def __init__(self):
         super(TinyModel, self).__init__()
 
-        self.layer1 = torch.nn.Linear(1000, 100)
+        self.layer1 = torch.nn.Linear(DIM_IN, HIDDEN_SIZE)
         self.relu = torch.nn.ReLU()
-        self.layer2 = torch.nn.Linear(100, 10)
+        self.layer2 = torch.nn.Linear(HIDDEN_SIZE, DIM_OUT)
 
     def forward(self, x):
         x = self.layer1(x)