File size: 146 Bytes
8056602
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
#!/bin/bash

GPU=${1}
CONFIG=${2}
ARGS=${@:3}

torchrun --standalone --nproc_per_node ${GPU} scripts/train_magicdrive.py \
     ${CONFIG} ${ARGS}