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.
45 lines
1.3 KiB
45 lines
1.3 KiB
#!/usr/bin/env bash |
|
|
|
# an example to do pre-training: (not that `/path/to/imagenet` should contain directories named `train` and `val`) |
|
# > cd /path/to/SparK |
|
# > bash ./main.sh experiment_name --num_nodes=1 --ngpu_per_node=8 --node_rank=0 --master_address=128.0.0.0 --master_port=30000 --data_path=/path/to/imagenet --model=convnext_small --ep=400 --wp_ep=10 |
|
|
|
####### template begins ####### |
|
SCRIPTS_DIR=$(cd $(dirname $0); pwd) |
|
cd "${SCRIPTS_DIR}" |
|
SPARK_DIR=$(pwd) |
|
echo "SPARK_DIR=${SPARK_DIR}" |
|
|
|
shopt -s expand_aliases |
|
alias python=python3 |
|
alias to_scripts_dir='cd "${SCRIPTS_DIR}"' |
|
alias to_spark_dir='cd "${SPARK_DIR}"' |
|
alias print='echo "$(date +"[%m-%d %H:%M:%S]") (exp.sh)=>"' |
|
function mkd() { |
|
mkdir -p "$1" >/dev/null 2>&1 |
|
} |
|
####### template ends ####### |
|
|
|
|
|
EXP_NAME=$1 |
|
|
|
EXP_DIR="${SPARK_DIR}/output_${EXP_NAME}" |
|
|
|
|
|
print "===================== Args =====================" |
|
print "EXP_NAME: ${EXP_NAME}" |
|
print "[other_args sent to launch.py]: ${*:2}" |
|
print "================================================" |
|
print "" |
|
|
|
|
|
print "============== Pretraining starts ==============" |
|
to_spark_dir |
|
touch ~/wait1 |
|
python launch.py \ |
|
--main_py_relpath main.py \ |
|
--exp_name "${EXP_NAME}" \ |
|
--exp_dir "${EXP_DIR}" \ |
|
"${*:2}" |
|
print "============== Pretraining ends ==============" |
|
rm ~/wait1
|
|
|