Update Yolov5_Deepsort/models/yolo.py

Yolov5_Deepsort/models/yolo.py

@@ -89,7 +89,13 @@ class Model(nn.Module):
             logger.info(f'Overriding model.yaml anchors with anchors={anchors}')
             self.yaml['anchors'] = round(anchors)  # override yaml value
         self.model, self.save = parse_model(deepcopy(self.yaml), ch=[ch])  # model, savelist
-        self.model = self.model.float()
+        for module in self.model.modules():
+            if hasattr(module, 'weight') and module.weight is not None:
+                module.weight.data = module.weight.data.float()
+            if hasattr(module, 'bias') and module.bias is not None:
+                module.bias.data = module.bias.data.float()
+
+        #self.model = self.model.float()
         self.names = [str(i) for i in range(self.yaml['nc'])]  # default names
         self.inplace = self.yaml.get('inplace', True)
         # logger.info([x.shape for x in self.forward(torch.zeros(1, ch, 64, 64))])