We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 0428745 + 8f31095 commit 65ed32cCopy full SHA for 65ed32c
extras/solutions/03_pytorch_computer_vision_exercise_solutions.ipynb
@@ -3598,7 +3598,7 @@
3598
"from tqdm.auto import tqdm\n",
3599
"from torchmetrics import Accuracy\n",
3600
"\n",
3601
- "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",
3602
3603
"# Setup training/testing loop\n",
3604
"epochs = 5\n",
0 commit comments