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向右移动来派生,因此您不必自己进行移动。

参考链接:

Glossary

What decoder_input_ids should be for sequence-to-sequence Transformer model?