transformers
Arde/fsdp activation checkpointing
#25771
Merged

Arde/fsdp activation checkpointing #25771

arde171
arde171 add FSDP config option to enable activation-checkpointing
b3b79b65
arde171 update docs
d1d6c668
HuggingFaceDocBuilderDev
ArthurZucker
pacman100
pacman100 commented on 2023-08-28
arde171 add checks and remove redundant code
d283fda9
arde171 fix formatting error
aad11bb9
pacman100
pacman100 approved these changes on 2023-08-29
pacman100 pacman100 merged 738ecd17 into main 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone