train_scripts/train.sh

#/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 \
        "$@"