IDEA-CCNL / GTS-Engine

GTS Engine: A powerful NLU Training System。GTS引擎(GTS-Engine)是一款开箱即用且性能强大的自然语言理解引擎,聚焦于小样本任务,能够仅用小样本就能自动化生产NLP模型。
https://gtsfactory.com/
Apache License 2.0
87 stars 9 forks source link

Dev liuyibo #37

Closed rookielyb closed 1 year ago

rookielyb commented 1 year ago

修复运行nli过程中的bug

rookielyb commented 1 year ago

1.AutoTokenizer 不兼容 "model_type"="megatron-bert"

企业微信截图_053e9827-98a3-49a8-a11a-e01d559443a7

2.解决句子对任务预测时报错的bug

企业微信截图_20b188e6-cea1-4803-b104-a7c36259af68

3.句子对任务增加threshold阈值参数

企业微信截图_79a6e8a3-8333-4a86-a1c7-1333e5843f66
rookielyb commented 1 year ago

1.sh run_train_qiankunding.sh 运行结果

企业微信截图_258ce686-c08b-4077-99ea-f6fe4e3e1798

2.sh run_inference_qiankunding.sh 运行结果

企业微信截图_120bf3d4-a512-420a-8cac-cbe126cc90c1