Commit 727e68c2 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Clarify checkpointing comment

parent f706ace4
......@@ -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
consumption but fewer checkpoints. If None, no checkpointing is
performed.
Size of each chunk. A higher value corresponds to fewer
checkpoints, and trades memory for speed. If None, no checkpointing
is performed.
Returns:
The output of the final block
"""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment