transformers
Fix multi gpu loss sync condition, add doc and test
#35743

Merged

Fix multi gpu loss sync condition, add doc and test #35743

SunMarc merged 4 commits into huggingface:main from techkang:main

techkang requested a review from

muellerzr 1 year ago

techkang requested a review from

SunMarc 1 year ago

techkang requested a review from

Rocketknight1 1 year ago

techkang requested a review from

ArthurZucker 1 year ago

muellerzr approved these changes on 2025-01-17

SunMarc approved these changes on 2025-01-17

SunMarc approved these changes on 2025-02-10

Fix multi gpu loss sync condition, add doc and test

e66bf221

rename function and class

308df55a

loss should not scale during inference

fa46bcea

fix typo

6b9eba92

techkang force pushed from 20dd133e to 6b9eba92 345 days ago

SunMarc merged a33ac830 into main 344 days ago

ArthurZucker commented on 2025-02-13

Reviewers

SunMarc

muellerzr

ArthurZucker

Rocketknight1

Assignees

No one assigned

Labels

None yet

Milestone

No milestone