YOLO-World3

Sleeping

stevengrove

initial commit

186701e 5 months ago

No virus

479 Bytes

	#!/usr/bin/env bash

	CONFIG=$1
	CHECKPOINT=$2
	GPUS=$3
	NNODES=${NNODES:-1}
	NODE_RANK=${NODE_RANK:-0}
	PORT=${PORT:-29500}
	MASTER_ADDR=${MASTER_ADDR:-"127.0.0.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}