Searched defs:apply_fsdp (Results 1 – 2 of 2) sorted by relevance
136 def apply_fsdp(args, model, use_checkpointing=False, use_wrap_policy=True): function
688 def apply_fsdp(model, wrap_policy): function