Update strong_sort/reid_multibackend.py

strong_sort/reid_multibackend.py

@@ -124,6 +124,8 @@ class ReIDDetectMultiBackend(nn.Module):
         return pt, jit, onnx, xml, engine, coreml, saved_model, pb, tflite, edgetpu, tfjs
     
     def warmup(self, imgsz=(1, 256, 128, 3)):
+        if isinstance(self.device, str):
+            self.device = torch.device(self.device if torch.cuda.is_available() else 'cpu')
         # Warmup model by running inference once
         warmup_types = self.pt, self.jit, self.onnx, self.engine, self.saved_model, self.pb
         if any(warmup_types) and self.device.type != 'cpu':