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

fixed division error: / -> //

Browse Source
This commit is contained in:
KaiyangZhou 2018-11-05 21:30:40 +00:00
parent 9726b8bb96
commit 81d3cc3995
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
torchreid/models/shufflenet.py
View File

@ -17,7 +17,7 @@ class ChannelShuffle(nn.Module):
def forward(self, x):
b, c, h, w = x.size()
n = c / self.g
n = c // self.g
# reshape
x = x.view(b, self.g, n, h, w)
# transpose