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'