transformers 在seq2seq 模型中的decoder_input_ids
Published on Aug. 22, 2023, 12:11 p.m.
decoder_input_ids 是来自 huggingface 的labels
decoder_input_ids(可选)对应标签,labels是提供decoder_input_ids的首选方式。
这是因为如果 decoder_input_ids 为 None,它们将通过将labels向右移动来派生,因此您不必自己进行移动。
参考链接:
What decoder_input_ids should be for sequence-to-sequence Transformer model?