run_garbage.sh 161 B

1234
  1. # -------------------------- Train Pipeline --------------------------
  2. python -m torch.distributed.run --nproc_per_node=1 --master_port 9000 garbage.py --cuda