Searched defs:use_checkpointing (Results 1 – 2 of 2) sorted by relevance
136 def apply_fsdp(args, model, use_checkpointing=False, use_wrap_policy=True): argument
181 def test(use_checkpointing, use_wrapper, use_reentrant): argument