change attribute name: last_linear -> classifier

pull/201/head
KaiyangZhou 2019-06-03 09:26:04 +01:00
parent 86ddbd97aa
commit 752cb82682
1 changed files with 2 additions and 2 deletions

View File

@ -308,7 +308,7 @@ class InceptionV4(nn.Module):
Inception_C()
)
self.global_avgpool = nn.AdaptiveAvgPool2d(1)
self.last_linear = nn.Linear(1536, num_classes)
self.classifier = nn.Linear(1536, num_classes)
def forward(self, x):
f = self.features(x)
@ -345,4 +345,4 @@ def inceptionv4(num_classes, loss='softmax', pretrained=True, **kwargs):
if pretrained:
model_url = pretrained_settings['inceptionv4']['imagenet']['url']
init_pretrained_weights(model, model_url)
return model
return model