From faf3aa7b3fda4946d8741f9763ce7584fe50e9fe Mon Sep 17 00:00:00 2001
From: khushpatel <61388622+khushpatelmd@users.noreply.github.com>
Date: Wed, 9 Jun 2021 13:42:18 -0500
Subject: [PATCH] Update 01_core.ipynb

Thank you for this amazing course.
I found one change
dim =1 not dim=0 for the softmax. I tried creating a smaller dummy example to understand it.
---
 course/training/01_core.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/course/training/01_core.ipynb b/course/training/01_core.ipynb
index 5dc4366..d7fc3c3 100644
--- a/course/training/01_core.ipynb
+++ b/course/training/01_core.ipynb
@@ -326,7 +326,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "softmax = torch.nn.functional.softmax(outputs.prediction_logits[0], dim=0)  # create probability distribution\n",
+    "softmax = torch.nn.functional.softmax(outputs.prediction_logits[0], dim=1)  # create probability distribution\n",
     "argmax = torch.argmax(softmax, dim=1)  # get index of the max probability"
    ]
   },