train_single_gpu.py 5.21 KB