# -------------------------- Train Pipeline -------------------------- python -m torch.distributed.run --nproc_per_node=${1} --master_port 9000 garbage.py --cuda