MedCat
/

table_detection_3

nguyenp99 commited on May 3

Commit

797a48f

•

1 Parent(s): 5bcdad5

Update stamp_processing/module/yolov5/yolo_utils/torch_utils.py

Files changed (1) hide show

stamp_processing/module/yolov5/yolo_utils/torch_utils.py CHANGED Viewed

@@ -9,7 +9,7 @@ import torch.nn.functional as F
 def select_device(device=""):
     cpu = device.lower() == "cpu"
     cuda = not cpu and torch.cuda.is_available()
-    return torch.device("cuda" if cuda else "cpu")
 def time_synchronized():

 def select_device(device=""):
     cpu = device.lower() == "cpu"
     cuda = not cpu and torch.cuda.is_available()
+    return torch.device("cuda:1" if cuda else "cpu")
 def time_synchronized():