ImportError:无法从'transformers'导入'TFTrainer'在Colab中

0 投票
1 回答
219 浏览
提问于 2025-04-12 22:24

我很困惑,为什么我在Colab里无法导入TFTraniner。

我试过:
!pip install TFTranier
!pip --upgrade transformers
还有重新安装transformers。

但是在Colab里还是无法导入TFTranier,就像截图里那样。
图片

我是在尝试跟着其他人的代码,他们成功地从transformers导入了TFTranier。
原始代码

1 个回答

0

TFTrainer是一个独立的包,你需要通过transformers来导入它。

这里有一个示例,教你如何从TFTrainer导入:

from tftrainer import TrainArgument, Trainer

https://github.com/Nam-SW/TFTrainer

补充说明

在最新版本的transformers中,tftrainer已经不再支持了,建议你使用旧版本。

我试过一个旧版本,它可以正常工作:

!pip install transformers==4.18.0

现在,像这样导入TFTrainer:

from transformers.trainer_tf import TFTrainer

撰写回答