update

BiomedSciAI · Aug 15, 2024 · dab7ae2 · dab7ae2
1 parent 342bc77
commit dab7ae2
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/fuse_examples/imaging/oai_example/downstream/classification.py b/fuse_examples/imaging/oai_example/downstream/classification.py
@@ -39,7 +39,7 @@
 def main(cfg: DictConfig) -> None:
     cfg = hydra.utils.instantiate(cfg)
     results_path = cfg.results_dir
-    os.environ["CUDA_VISIBLE_DEVICES"] = ",".join(cfg.cuda_devices)
+    os.environ["CUDA_VISIBLE_DEVICES"] = ",".join([str(x) for x in cfg.cuda_devices])
 
     assert (
         sum(

diff --git a/fuse_examples/imaging/oai_example/downstream/segmentation.py b/fuse_examples/imaging/oai_example/downstream/segmentation.py
@@ -35,7 +35,7 @@
 def main(cfg: DictConfig) -> None:
     cfg = hydra.utils.instantiate(cfg)
     results_path = cfg.results_dir
-    os.environ["CUDA_VISIBLE_DEVICES"] = ",".join(cfg.cuda_devices)
+    os.environ["CUDA_VISIBLE_DEVICES"] = ",".join([str(x) for x in cfg.cuda_devices])
 
     assert (
         sum(

diff --git a/fuse_examples/imaging/oai_example/self_supervised/dino.py b/fuse_examples/imaging/oai_example/self_supervised/dino.py
@@ -157,7 +157,7 @@ def configure_optimizers(self) -> tuple:
 @hydra.main(version_base="1.2", config_path=".", config_name="dino")
 def main(cfg: DictConfig) -> None:
     cfg = hydra.utils.instantiate(cfg)
-    os.environ["CUDA_VISIBLE_DEVICES"] = ",".join(cfg.cuda_devices)
+    os.environ["CUDA_VISIBLE_DEVICES"] = ",".join([str(x) for x in cfg.cuda_devices])
 
     assert (
         sum(