You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
11 lines
320 B
11 lines
320 B
#!/usr/bin/env bash
|
|
# Copyright (c) OpenMMLab. All rights reserved.
|
|
|
|
CONFIG=$1
|
|
CHECKPOINT=$2
|
|
GPUS=$3
|
|
PORT=${PORT:-29500}
|
|
|
|
PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
|
|
python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \
|
|
$(dirname "$0")/test.py $CONFIG $CHECKPOINT --launcher pytorch ${@:4}
|
|
|