IDSF-JointBERT_CRF / run_jointBERT-CRF_PhoBERTencoder.sh
DD0101's picture
first commit
c938124
raw
history blame
776 Bytes
export lr=3e-5
export c=0.6
export s=100
echo "${lr}"
export MODEL_DIR=JointBERT-CRF_PhoBERTencoder
export MODEL_DIR=$MODEL_DIR"/"$lr"/"$c"/"$s
echo "${MODEL_DIR}"
python3 main.py --token_level word-level \
--model_type phobert \
--model_dir $MODEL_DIR \
--data_dir PhoATIS \
--seed $s \
--do_train \
--do_eval \
--save_steps 140 \
--logging_steps 140 \
--num_train_epochs 50 \
--tuning_metric mean_intent_slot \
--use_crf \
--gpu_id 0 \
--embedding_type soft \
--intent_loss_coef $c \
--learning_rate $lr