From 60814eea0810b5734481dba41f71675ba8976764 Mon Sep 17 00:00:00 2001 From: Tal Date: Tue, 10 Dec 2024 08:44:18 +0200 Subject: [PATCH] Update tresnet.py --- src/models/tresnet/tresnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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):