transformers
2b4e2b3e - Let gradient checkpointing skip layers with every_n_layers (#48200)

Commit
3 days ago
Let gradient checkpointing skip layers with every_n_layers (#48200) * Let gradient checkpointing skip layers with every_n_layers * Read every_n_layers from gradient_checkpointing_kwargs instead of a new argument * Document every_n_layers in the field help text too --------- Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com>
Author
Parents
Loading