updating a typo :-)

pytorch · Jul 3, 2023 · 768f32d · 768f32d
1 parent d06d2f1
commit 768f32d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/beginner_source/introyt/trainingyt.py b/beginner_source/introyt/trainingyt.py
@@ -148,7 +148,7 @@ def forward(self, x):
         x = x.view(-1, 16 * 4 * 4)
         x = F.relu(self.fc1(x))
         x = F.relu(self.fc2(x))
-        x = self.fc3(x)
+        x = F.softmax(self.fc3(x))
         return x