change attribute name: last_linear -> classifier
parent
86ddbd97aa
commit
752cb82682
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue