fix device string

MIC-DKFZ · Oct 16, 2024 · 7524715 · 7524715
1 parent c9dc844
commit 7524715
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/HD_BET/run.py b/HD_BET/run.py
@@ -61,7 +61,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)