[🐛 BUG] Fix bugs when collecting results from mp.spawn in multi-GPU training

[🐛 BUG] Fix bugs when collecting results from mp.spawn in multi-GPU training #1204