fix device string

MIC-DKFZ · Oct 18, 2024 · 679d774 · 679d774
1 parent bc99f8f
commit 679d774
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/HD_BET/run.py b/HD_BET/run.py
@@ -66,7 +66,7 @@ def run_hd_bet(mri_fnames, output_fnames, mode="accurate", config_file=os.path.j
     if device == 'cpu' or device == 'mps':
         pass
     else:
-        device = "cuda:" + device
+        device = "cuda:" + str(device)
 
     net = net.to(device)