accelerate
allow gather_for_metrics to be more flexible
#2710
Merged

Loading