diff --git a/vital/config/trainer/default.yaml b/vital/config/trainer/default.yaml index 6a3226014..e7f29fe3a 100644 --- 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 index e22a49261..1955b98f9 100644 --- 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)