Update Yolov5_Deepsort/models/common.py

Yolov5_Deepsort/models/common.py

@@ -35,8 +35,8 @@ class Conv(nn.Module):
     def __init__(self, c1, c2, k=1, s=1, p=None, g=1, act=True):  # ch_in, ch_out, kernel, stride, padding, groups
         super(Conv, self).__init__()
         self.conv = nn.Conv2d(c1, c2, k, s, autopad(k, p), groups=g, bias=False)
-        self.bn = nn.BatchNorm2d(c2)
-        self.act = nn.SiLU() if act is True else (act if isinstance(act, nn.Module) else nn.Identity())
+        self.bn = nn.BatchNorm2d(c2).float()
+        self.act = nn.SiLU().float() if act is True else (act if isinstance(act, nn.Module) else nn.Identity())
 
     def forward(self, x):
         x = x.float()