machine-learning-deep-learning/Machine-Learning-Collection
1
0
Fork 0
mirror of https://github.com/aladdinpersson/Machine-Learning-Collection.git synced 2026-02-20 13:50:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #29 from darveenvijayan/patch-1

Browse Source 
Update pytorch_inceptionet.py
This commit is contained in:
Aladdin Persson
2021-03-24 13:04:45 +01:00
committed by GitHub
parent 83d234daec 1ba55b7382
commit 1b761f1443
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ML/Pytorch/CNN_architectures/pytorch_inceptionet.py
View File

@@ -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)