single-GPU training

[YusenZheng]

Hello, this code uses distributed multi-GPU training in multiple places, but I only have a single GPU. When running this code, I encounter multiple errors. Could you provide a version of the code for single-GPU training?