diff --git a/src/models/tresnet/tresnet.py b/src/models/tresnet/tresnet.py
index c086582..234bb78 100644
--- a/src/models/tresnet/tresnet.py
+++ b/src/models/tresnet/tresnet.py
@@ -44,7 +44,7 @@ def __init__(self, inplanes, planes, stride=1, downsample=None, use_se=True, ant
         self.relu = nn.ReLU(inplace=True)
         self.downsample = downsample
         self.stride = stride
-        reduce_layer_planes = max(planes * self.expansion // 4, 64)
+        reduce_layer_planes = max(planes * self.expansion / 4, 64)
         self.se = SEModule(planes * self.expansion, reduce_layer_planes) if use_se else None
 
     def forward(self, x):