Transformer:导入包时出错。“导入错误:无法从‘torch.optim.lr_scheduler’导入名称‘SAVE_STATE_WARNING’”
我正在 Google Colab 上进行机器学习项目,最近似乎在尝试从转换器导入包时出现问题。错误消息说:
导入错误:无法从“torch.optim.lr_scheduler”(/usr/local/lib/python3.7/dist-packages/torch/optim/lr_scheduler.py)导入名称“SAVE_STATE_WARNING”
代码很简单,如下所示:
!pip install transformers==3.5.1
from transformers import BertTokenizer
到目前为止,我已经尝试安装不同版本的转换器,并导入一些其他包,但似乎导入任何包:
from transformers import *Package
不工作,并会导致同样的错误。我想知道是否有人也遇到同样的问题?
回答
通过运行此命令在 colab 中更改火炬版本
!pip install torch==1.4.0。然后,它对我有用。
THE END
二维码