问题描述
我正在尝试使用 Google 的 T5 进行语言翻译。但是,它不适用于德语到英语。
英语到德语的工作正常:
self.tokenizer = AutoTokenizer.from_pretrained("t5-small")
self.model = AutoModelForSeq2SeqLM.from_pretrained("t5-small")
inputs = self.tokenizer.encode("translate English to German: " + text,return_tensors="pt",max_length=512,truncation=True)
summary_ids = self.model.generate(inputs,min_length=5,length_penalty=5.,num_beams=2)
summary = self.tokenizer.decode(summary_ids[0])
但是,将编码更改为“德语到英语”不起作用。
这个模型是不是为了将德语翻译成英语,还是我用错了?
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)