Transformer：导入包时出错。“导入错误：无法从‘torch.optim.lr_scheduler’导入名称‘SAVE_STATE_WARNING’”

Question

我正在 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

不工作，并会导致同样的错误。我想知道是否有人也遇到同样的问题？

Answer 1

通过运行此命令在 colab 中更改火炬版本

!pip install torch==1.4.0。然后，它对我有用。