1 Star 0 Fork 0

夏周睿/JointIDSF

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
run_jointIDSF_PhoBERTencoder.sh 1.11 KB
一键复制 编辑 原始数据 按行查看 历史
joey234 提交于 2021-06-03 07:26 . Update data path
#As we initialize JointIDSF from JointBERT, user need to train a base model JointBERT first
./run_jointBERT-CRF_PhoBERTencoder.sh
#Train JointIDSF
export lr=4e-5
export c=0.15
export s=100
echo "${lr}"
export MODEL_DIR=JointIDSF_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_intent_context_attention \
--attention_embedding_size 200 \
--use_crf \
--gpu_id 0 \
--embedding_type soft \
--intent_loss_coef $c \
--pretrained \
--pretrained_path JointBERT-CRF_PhoBERTencoder/3e-5/0.6/100 \
--learning_rate $lr
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/CharlieShark/JointIDSF.git
git@gitee.com:CharlieShark/JointIDSF.git
CharlieShark
JointIDSF
JointIDSF
main

搜索帮助

D67c1975 1850385 1daf7b77 1850385