File size: 417 Bytes
c614b0f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
ACC_CONFIG="./configs/accelerate-train.yaml"
TRAIN_CONFIG="./configs/train-sample-human.yaml"
if [ -n "$1" ]; then
TRAIN_CONFIG=$1
else
TRAIN_CONFIG="./configs/train-sample-human.yaml"
fi
if [ -n "$2" ]; then
MAIN_PORT=$2
else
MAIN_PORT=12345
fi
accelerate launch --config_file $ACC_CONFIG --main_process_port=$MAIN_PORT -m openlrm.launch train.human_lrm --config $TRAIN_CONFIG |