forked from VinAIResearch/JointIDSF
-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_jointIDSF_XLM-Rencoder.sh
executable file
·30 lines (30 loc) · 1.1 KB
/
run_jointIDSF_XLM-Rencoder.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
#As we initialize JointIDSF from JointBERT, user need to train a base model JointBERT first
./run_jointBERT-CRF_XLM-Rencoder.sh
#Train JointIDSF
export lr=3e-5
export c=0.25
export s=10
echo "${lr}"
export MODEL_DIR=JointIDSF_XLM-Rencoder
export MODEL_DIR=$MODEL_DIR"/"$lr"/"$c"/"$s
echo "${MODEL_DIR}"
python3 main.py --token_level syllable-level \
--model_type xlmr \
--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_intent_context_attention \
--attention_embedding_size 200 \
--use_crf \
--gpu_id 0 \
--embedding_type soft \
--intent_loss_coef $c \
--pretrained \
--pretrained_path JointBERT-CRF_XLM-Rencoder/4e-5/0.45/10 \
--learning_rate $lr