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

Update pytorch_inceptionet.py

Browse Source
This commit is contained in:
Darveen Vijayan
2021-02-10 19:14:55 +08:00
committed by GitHub
parent 2d876db44c
commit 1ba55b7382
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)