Searched defs:wrap_policy (Results 1 – 1 of 1) sorted by relevance
167 model, wrap_policy, checkpoint_policy, use_activation_checkpointing=True argument688 def apply_fsdp(model, wrap_policy): argument