mirrors/deep-person-reid
1
0
Fork 0
mirror of https://github.com/KaiyangZhou/deep-person-reid.git synced 2025-06-03 14:53:23 +08:00
Code Issues Projects Releases Wiki Activity

add eval mode

Browse Source
This commit is contained in:
KaiyangZhou 2018-05-12 17:45:16 +01:00
parent e8c3f95c15
commit cc99c36bbf
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
models/SqueezeNet.py
View File

@ -108,6 +108,9 @@ class SqueezeNet(nn.Module):
x10 = F.relu(self.conv10(x9))
f = F.avg_pool2d(x10, x10.size()[2:]).view(x10.size(0), -1)
if not self.training:
return f
if self.loss == {'xent'}:
return f
elif self.loss == {'xent', 'htri'}: