【报错解决方案】TypeError :_set_gradient_checkpointing()got an unexpected keyword argument ‘enable‘

我在模型训练的过程中遇到了这样一个问题:
File “/opt/conda/lib/python3.9/site-packages/transformers/modeling_utils.py", line 1159, in post_init self._backward_compatibility_gradient_checkpointing()
File "/opt/conda/lib/python3.9/site-packages/transformers/modeling_utils.py", line 1163,in _backward_compatibility_gradient_checkpointing self.gradient_checkpointing_enable()
File "/opt/conda/lib/python3.9/site-packages/transformers/modeling_utils.py", line 1872, in gradient_checkpointing_enable self._set_gradient_checkpointing(enable=True, gradient_checkpointing_func=gradient_checkpointing_func) TypeError :_set_gradient_checkpointing()got an unexpected keyword argument 'enable'

这个报错显示的问题是没有“enable”这个模块

但是我在查询pip show enable的时候发现我是有enable模块的,所以问题可能是因为其他的问题导致了enable模块无法被读取。

在全网查询了这个错误发现没有解决方案。

我就查看了一些基础模块的版本,发现transformers的版本非常新(目前是4.35.x),但是我们知道项目中对transformers的版本的要求是非常严格的。

查询官方文档发现很多项目对transformers的版本下调会提高模块的兼容性。

所以尝试之后发现transformers的版本下降到4.31.0就可以解决“enable”模块无法识别到的问题

pip install transformers==4.31.0

  • 18
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值