checkpointing.refactor