PR #29657 Fix TPU checkpointing inside Trainer

Fix TPU checkpointing inside Trainer #29657

amyeroberts merged 1 commit into huggingface:main from shub-kris:fix/trainer-checkpointing-on-tpu

Manually call sync step

703f7708

shub-kris changed the title ~~Manually call sync step~~ Fix TPU checkpointing inside Trainer 2 years ago

shub-kris requested a review from

muellerzr 2 years ago

shub-kris requested a review from

pacman100 2 years ago

muellerzr approved these changes on 2024-03-14

pacman100 approved these changes on 2024-03-14

pacman100 requested a review from

amyeroberts 2 years ago

amyeroberts approved these changes on 2024-03-14

amyeroberts merged 956f44f1 into main 2 years ago

Reviewers

amyeroberts

pacman100

muellerzr

Assignees

No one assigned

Labels

None yet

Milestone

No milestone