transformers
Warn on TPUs when the custom optimizer and model device are not the same
#18668

Merged

Warn on TPUs when the custom optimizer and model device are not the same #18668

muellerzr merged 2 commits into main from warn_tpu

Check optimizer for device on TPU

bcca6802

Typo

58dc9297

muellerzr added Distributed Training / Models

muellerzr added bug

muellerzr requested a review from

LysandreJik 3 years ago

muellerzr requested a review from

sgugger 3 years ago

sgugger approved these changes on 2022-08-31

muellerzr merged e4910213 into main 3 years ago

muellerzr deleted the warn_tpu branch 3 years ago

Reviewers

sgugger

LysandreJik

Assignees

No one assigned

Labels

bug Distributed Training / Models

Milestone

No milestone