Describe the bug
When doing distributed training, ClassificationSaver runs on different ranks and write to the same CSV file, it causes a random output in the file and sometimes even only 1 rank can open the file.
Expected behavior
Expect all_gather image path and model output data, then only write CSV file in the specified rank.