File size: 320 Bytes
0a7036f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | START_PORT=${START_PORT:-29056}
MASTER_PORT=${MASTER_PORT:-29061}
save_root='visualization/'
mkdir -p $save_root
python -m torch.distributed.run \
--nproc_per_node 1 \
--master_port $MASTER_PORT \
wan_va/wan_va_server.py \
--config-name robotwin \
--port $START_PORT \
--save_root $save_root
|