accelerate

0a16f37b - Ensure that last batch doesn't get dropped if perfectly even in gather_for_metrics (#982)