Update Yolov5_Deepsort/models/common.py

Yolov5_Deepsort/models/common.py

@@ -234,6 +234,7 @@ class AutoShape(nn.Module):
     def __init__(self, model):
         super(AutoShape, self).__init__()
         self.model = model.eval()
+        self.model = self.model.float()
 
     def autoshape(self):
         print('AutoShape already enabled, skipping... ')  # model already converted to model.autoshape()