| #!/usr/bin/env bash |
| CONFIG=configs/eval_text.py |
| CHECKPOINT=work_dirs/anymo_stage_3_3B_210e_48gpu/e210.pth |
| GPUS=1 |
| NNODES=${NNODES:-1} |
| NODE_RANK=${NODE_RANK:-0} |
| PORT=${PORT:-29510} |
| export PORT=29501 |
| |
| MASTER_ADDR=${MASTER_ADDR:-"127.0.0.1"} |
| export CUDA_LAUNCH_BLOCKING=1 |
| PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \ |
| python -m torch.distributed.launch \ |
| --nnodes=$NNODES \ |
| --node_rank=$NODE_RANK \ |
| --master_addr=$MASTER_ADDR \ |
| --nproc_per_node=$GPUS \ |
| --master_port=$PORT \ |
| $(dirname "$0")/test.py \ |
| $CONFIG \ |
| $CHECKPOINT \ |
| --launcher pytorch \ |
| ${@:4} |
|
|
| python gpu.py |
|
|
