Clarify checkpointing comment

727e68c2 · Gustaf Ahdritz · f706ace4 · 727e68c2
Commit 727e68c2 authored Oct 22, 2021 by Gustaf Ahdritz
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

openfold/utils/checkpointing.py openfold/utils/checkpointing.py +3 -3

No files found.
--- a/openfold/utils/checkpointing.py
+++ b/openfold/utils/checkpointing.py
@@ -39,9 +39,9 @@ def checkpoint_blocks(
        args:
            Tuple of arguments for the first block.
        blocks_per_ckpt:
-            Size of each chunk. A higher value corresponds to higher memory
+            Size of each chunk. A higher value corresponds to fewer 
-            consumption but fewer checkpoints. If None, no checkpointing is
+            checkpoints, and trades memory for speed. If None, no checkpointing 
-            performed.
+            is performed.
    Returns:
        The output of the final block
    """