Migrate from gpus flag (to be deprecated in PL 1.7) to `accelerator…

…` and `devices`
vitalab · Aug 30, 2022 · 92a243b · 92a243b
1 parent 101c212
commit 92a243b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/vital/config/trainer/default.yaml b/vital/config/trainer/default.yaml
@@ -1,3 +1,4 @@
 _target_: pytorch_lightning.Trainer
-gpus: ${sys.gpus:}
+accelerator: "auto"
+devices: null
 enable_progress_bar: True
diff --git a/vital/utils/sys.py b/vital/utils/sys.py
@@ -9,7 +9,6 @@
 
 def register_omegaconf_resolvers() -> None:
     """Registers various OmegaConf resolvers useful to query system info."""
-    OmegaConf.register_new_resolver("sys.gpus", lambda x=None: int(torch.cuda.is_available()))
     OmegaConf.register_new_resolver("sys.num_workers", lambda x=None: os.cpu_count() - 1)
     OmegaConf.register_new_resolver("sys.getcwd", lambda x=None: os.getcwd())
     OmegaConf.register_new_resolver("sys.eps.np", lambda dtype: np.finfo(np.dtype(dtype)).eps)