mirrors
/
deep-person-reid
mirror of https://github.com/KaiyangZhou/deep-person-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
torchreid/models/inceptionv4.py
View File

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