Backport PR scverse#1473: pin memory False!

meeseeksmachine · Mar 29, 2022 · 8c65e7e · 8c65e7e
1 parent eed6271
commit 8c65e7e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/scvi/_settings.py b/scvi/_settings.py
@@ -36,6 +36,10 @@ class ScviConfig:
     >>> import logging
     >>> scvi.settings.verbosity = logging.INFO
 
+    To set pin memory for GPU training
+
+    >>> scvi.settings.dl_pin_memory_gpu_training = True
+
     To set the number of threads PyTorch will use
 
     >>> scvi.settings.num_threads = 2
@@ -53,7 +57,7 @@ def __init__(
         seed: int = 0,
         logging_dir: str = "./scvi_log/",
         dl_num_workers: int = 0,
-        dl_pin_memory_gpu_training: bool = True,
+        dl_pin_memory_gpu_training: bool = False,
         jax_preallocate_gpu_memory: bool = False,
     ):