Merge pull request #938 from Hrut7hesh/patch-1

Update 03_pytorch_computer_vision_exercise_solutions.ipynb
mrdbourke · Aug 22, 2024 · 65ed32c · 65ed32c
2 parents 0428745 + 8f31095
commit 65ed32c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/extras/solutions/03_pytorch_computer_vision_exercise_solutions.ipynb b/extras/solutions/03_pytorch_computer_vision_exercise_solutions.ipynb
@@ -3598,7 +3598,7 @@
         "from tqdm.auto import tqdm\n",
         "from torchmetrics import Accuracy\n",
         "\n",
-        "acc_fn = Accuracy(num_classes=len(fashion_mnist_class_names)).to(device) \n",
+        "acc_fn = Accuracy(task = 'multiclass', num_classes=len(fashion_mnist_class_names)).to(device) \n",
         "\n",
         "# Setup training/testing loop\n",
         "epochs = 5\n",