From 9596b6e5ae56249c9ffe518b9c62cb8d682281d8 Mon Sep 17 00:00:00 2001 From: Koos du Preez <35713435+kooscode@users.noreply.github.com> Date: Sat, 14 Nov 2020 15:43:32 -0800 Subject: [PATCH] Update image_classification_using_transfer_learning_in_pytorch.ipynb Num classes should not be entries -1 --- ...mage_classification_using_transfer_learning_in_pytorch.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Image-Classification-in-PyTorch/image_classification_using_transfer_learning_in_pytorch.ipynb b/Image-Classification-in-PyTorch/image_classification_using_transfer_learning_in_pytorch.ipynb index d66db45b4..9346b8320 100644 --- a/Image-Classification-in-PyTorch/image_classification_using_transfer_learning_in_pytorch.ipynb +++ b/Image-Classification-in-PyTorch/image_classification_using_transfer_learning_in_pytorch.ipynb @@ -83,7 +83,7 @@ "bs = 32\n", "\n", "# Number of classes\n", - "num_classes = len(os.listdir(valid_directory))-1 #10#2#257\n", + "num_classes = len(os.listdir(valid_directory)) #10#2#257\n", "print(num_classes)\n", "\n", "# Load Data from folders\n",