From 08611b0d31eb3a63531aff9630bbecf55b4b08de Mon Sep 17 00:00:00 2001 From: Irene Dea Date: Wed, 25 Oct 2023 19:21:15 -0700 Subject: [PATCH] Make default for cuda_load_lazy false (#694) --- scripts/train/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/train/train.py b/scripts/train/train.py index 8c1c28eb5c..e29f2c9a47 100644 --- a/scripts/train/train.py +++ b/scripts/train/train.py @@ -219,7 +219,7 @@ def main(cfg: DictConfig) -> Trainer: # Set CUDA lazy loading # This can save a bit of memory if not all modules are needed - cuda_load_lazy: bool = cfg.pop('cuda_load_lazy', True) + cuda_load_lazy: bool = cfg.pop('cuda_load_lazy', False) if cuda_load_lazy: os.environ['CUDA_MODULE_LOADING'] = 'LAZY'