| #/bin/bash | |
| set -e | |
| work_dir=output/debug | |
| np=8 | |
| if [[ $1 == *.yaml ]]; then | |
| config=$1 | |
| shift | |
| else | |
| config="configs/sana_config/512ms/sample_dataset.yaml" | |
| echo "Only support .yaml files, but get $1. Set to --config_path=$config" | |
| fi | |
| TRITON_PRINT_AUTOTUNING=1 \ | |
| torchrun --nproc_per_node=$np --master_port=15432 \ | |
| train_scripts/train.py \ | |
| --config_path=$config \ | |
| --work_dir=$work_dir \ | |
| --name=tmp \ | |
| --resume_from=latest \ | |
| --report_to=tensorboard \ | |
| --debug=true \ | |
| "$@" | |