mirror of
https://github.com/open-mmlab/mmselfsup.git
synced 2025-06-03 14:59:38 +08:00
10 lines
256 B
Bash
10 lines
256 B
Bash
|
#!/usr/bin/env bash
|
||
|
|
|
||
|
|
CONFIG=$1
|
||
|
|
GPUS=$2
|
||
|
|
PORT=${PORT:-29500}
|
||
|
|
|
||
|
|
PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
|
||
|
|
python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \
|
||
|
|
$(dirname "$0")/train.py $CONFIG --seed 0 --launcher pytorch ${@:3}
|