From 81d3cc39952929d24c6d703872fdbbc3ce10c42c Mon Sep 17 00:00:00 2001 From: KaiyangZhou Date: Mon, 5 Nov 2018 21:30:40 +0000 Subject: [PATCH] fixed division error: / -> // --- torchreid/models/shufflenet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchreid/models/shufflenet.py b/torchreid/models/shufflenet.py index fd005db..3d73d38 100644 --- a/torchreid/models/shufflenet.py +++ b/torchreid/models/shufflenet.py @@ -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