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