PR #24389 [Trainer] Fix optimizer step on PyTorch TPU

[Trainer] Fix optimizer step on PyTorch TPU #24389

sgugger merged 2 commits into huggingface:main from cowanmeg:trainer-xla-optimizer-fix

update optimizer step for tpu

ca91be49

add comment

03c852ea

muellerzr approved these changes on 2023-06-21

sgugger approved these changes on 2023-06-21

sgugger merged 1815d186 into main 3 years ago

Reviewers

sgugger

muellerzr

Assignees

No one assigned

Labels

None yet

Milestone

No milestone