t5-pegasus-pytorch
t5-pegasus-pytorch copied to clipboard
请问如何取消T5模型生成的开始标记?
在T5decoder的输出中,输出的logits是一个表示开始的special token。由于我的任务的特殊性,只需要生成一个toeken,所以这个开始标记会影响结果。所以我想知道,在T5中可不可以直接生成target token。
在T5decoder的输出中,输出的logits是一个表示开始的special token。由于我的任务的特殊性,只需要生成一个toeken,所以这个开始标记会影响结果。所以我想知道,在T5中可不可以直接生成target token。