如何自定义fairseq模型输入?

2024-06-14 20:42:07 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在使用pytorch/fairseq创建一个seq2seq翻译模型,该模型使用srcèu标记及其postag作为输入,以改进机器翻译结果,其方式如paper所述。我已经习惯了预处理.py文件来预处理标记文件,但仍然混淆了fairseq模型如何接收输入。请解释一下fairseq模型是如何接收输入的,并给我一些关于如何让模型接收postag和src令牌作为张量的想法。谢谢


Tags: 文件py标记模型src方式pytorchpaper