Update src/htr_pipeline/utils/filter_segmask.py

src/htr_pipeline/utils/filter_segmask.py

@@ -7,7 +7,7 @@ from mmengine.structures import InstanceData
 
 class FilterSegMask:
     def __init__(self):
-        self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+        self.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
     # Removes smaller masks that are contained in a bigger mask
     # @timer_func