Module strategy
Expand description
CheckpointStrategy module
Structs§
- Balanced
Checkpointing - Operation properties are as they are marked (compute or memory bound)
- NoCheckpointing
- All operations are considered compute bound, notwithstanding how they are marked
Enums§
- Checkpointing
Error - Error that can happen when trying to checkpoint a tensor.
Traits§
- Checkpoint
Strategy - Strategy for the amount of checkpointing to do during autodiff