unilm [TrOCR] 如何使用自定义的不同语言数据集训练TrOCR?

hm2xizp9  于 2个月前  发布在  其他
关注(0)|答案(2)|浏览(48)

你好,
我想将TRocr的解码器部分更改为在不同的词汇表(即不同的语言)上进行训练和推断。我正在遵循here的示例实现,但我无法更改词汇表,只能更改词汇表的大小。如果能告诉我如何更改TRocr解码器部分的词汇表,那将非常有帮助。
谢谢!

相关问题