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