Merge pull request aladdinpersson#29 from darveenvijayan/patch-1

Update pytorch_inceptionet.py

Author: aladdinpersson

ML/Pytorch/CNN_architectures/pytorch_inceptionet.py

@@ -51,7 +51,7 @@ def __init__(self, aux_logits=True, num_classes=1000):
 
  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)