HUSTAI / uie_pytorch

PaddleNLP UIE模型的PyTorch版实现
Apache License 2.0
539 stars 93 forks source link

uie pytorch config 参数问题 #4

Open shuopwang opened 1 year ago

shuopwang commented 1 year ago

在uie-base的config.json中,没有task_id的值,所以实际运行过程,task_type_embeddings没有生效?一直采用默认值0吗?

heiheiyoyo commented 1 year ago

task_type_embeddings是百度ERNIE模型在预训练时用到的,用于区分预训练任务类型,在微调和预测时都用默认值0