From 1ba55b73824a4eb1c0c30c5afd59bfda6f715edc Mon Sep 17 00:00:00 2001
From: Darveen Vijayan <43519646+darveenvijayan@users.noreply.github.com>
Date: Wed, 10 Feb 2021 19:14:55 +0800
Subject: [PATCH] Update pytorch_inceptionet.py

---
 ML/Pytorch/CNN_architectures/pytorch_inceptionet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ML/Pytorch/CNN_architectures/pytorch_inceptionet.py b/ML/Pytorch/CNN_architectures/pytorch_inceptionet.py
index c70cde7..38faef9 100644
--- a/ML/Pytorch/CNN_architectures/pytorch_inceptionet.py
+++ b/ML/Pytorch/CNN_architectures/pytorch_inceptionet.py
@@ -51,7 +51,7 @@ class GoogLeNet(nn.Module):
 
         self.avgpool = nn.AvgPool2d(kernel_size=7, stride=1)
         self.dropout = nn.Dropout(p=0.4)
-        self.fc1 = nn.Linear(1024, 1000)
+        self.fc1 = nn.Linear(1024, num_classes)
 
         if self.aux_logits:
             self.aux1 = InceptionAux(512, num_classes)