PR #22954 Add gradient checkpointing to Whisper Flax

Add gradient checkpointing to Whisper Flax #22954

sgugger merged 2 commits into huggingface:main from versae:add-gradient-checkpointing-whisper-flax

Add gradient checkpointing to Whisper Flax

c68582a9

versae marked this pull request as ready for review 2 years ago

sanchit-gandhi approved these changes on 2023-04-25

self.gradient_checkpointing only needed in nn.Module, removing unnece…

d3ecfdfe

sanchit-gandhi requested a review from

sgugger 2 years ago

sgugger approved these changes on 2023-04-26

sgugger merged ba0dc545 into main 2 years ago

versae deleted the add-gradient-checkpointing-whisper-flax branch 2 years ago

Reviewers

sgugger

sanchit-gandhi

Assignees

No one assigned

Labels

None yet

Milestone

No milestone