print_cuda_memory_usage docstring

2024-04-10 16:03:39 +00:00
parent daecc3b64c
commit 0f0113a7a6
2 changed files with 8 additions and 8 deletions
--- a/lerobot/common/utils.py
+++ b/lerobot/common/utils.py
@@ -98,6 +98,7 @@ def init_hydra_config(config_path: str, overrides: list[str] | None = None) -> D


 def print_cuda_memory_usage():
+    """Use this function to locate and debug memory leak."""
    import gc

    gc.collect()