mirror of
https://github.com/open-mmlab/mmselfsup.git
synced 2025-06-03 14:59:38 +08:00
10 lines
186 B
Bash
10 lines
186 B
Bash
#!/bin/bash
|
|
PYTHON=${PYTHON:-"python"}
|
|
|
|
CFG=$1
|
|
PY_ARGS=${@:2}
|
|
|
|
WORK_DIR=$(echo ${CFG%.*} | sed -e "s/configs/work_dirs/g")/
|
|
|
|
$PYTHON -u tools/train.py $1 --work_dir $WORK_DIR ${PY_ARGS}
|