accelerate
Ensure that last batch doesn't get dropped if perfectly even in gather_for_metrics
#982

Merged

Ensure that last batch doesn't get dropped if perfectly even in gather_for_metrics #982

muellerzr merged 2 commits into main from test-last-batch

Add test_last_batch

02353661

Fix gather bug

40ff8656

muellerzr added enhancement

muellerzr requested a review from

sgugger 3 years ago

sgugger approved these changes on 2023-01-18

muellerzr merged 0a16f37b into main 3 years ago

muellerzr deleted the test-last-batch branch 3 years ago

Reviewers

sgugger

Assignees

No one assigned

Labels

enhancement

Milestone

No milestone