时间戳

首页 > 解决方案 > 在哪里可以找到要导入项目的 Seq2SeqTrainer？

python - 在哪里可以找到要导入项目的 Seq2SeqTrainer？

问题描述

正如标题所说，我的项目需要 Seq2SeqTrainer，但 Github 上的文件不可用并返回 404。我使用此代码尝试导入它：

!wget https://raw.githubusercontent.com/huggingface/transformers/master/examples/seq2seq/seq2seq_trainer.py

如果它在任何地方都不再可用，那么什么是一个不错的选择，因为我也想使用 ROUGE。

标签： pythonnlpseq2seq

解决方案

我最终找到了解决方案。该文件位于： https ://github.com/huggingface/transformers/blob/master/examples/legacy/seq2seq/seq2seq_trainer.py

出于某种原因，在导入文件时，Python 会选择一个注释链接并引发错误。要解决这个问题，只需复制文件，顶部不带注释。这对我有用。

编辑：我找到了一个更整洁的解决方案：

from transformers import Seq2SeqTrainer

推荐阅读