Update Yolov5_Deepsort/models/common.py

Yolov5_Deepsort/models/common.py

@@ -34,7 +34,7 @@ class Conv(nn.Module):
     # Standard convolution
     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.conv = nn.Conv2d(c1, c2, k, s, autopad(k, p), groups=g, bias=False).float()
         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())