请问如何取消T5模型生成的开始标记？

renmada / t5-pegasus-pytorch

400 stars 60 forks source link

Open estuday opened 1 year ago

estuday commented 1 year ago

在T5decoder的输出中，输出的logits是一个表示开始的special token。由于我的任务的特殊性，只需要生成一个toeken，所以这个开始标记会影响结果。所以我想知道，在T5中可不可以直接生成target token。

renmada commented 1 year ago

可以自行修改tokenizer相关代码