尝试导入warmup_linear
时,出现此错误
ImportError: cannot import name 'warmup_linear'
导入-
from pytorch_pretrained_bert.optimization import BertAdam, warmup_linear
需求文件
boto3==1.9.198
botocore==1.12.198
certifi==2019.6.16
chardet==3.0.4
docutils==0.14
h5py==2.9.0
idna==2.8
jmespath==0.9.4
Keras==2.2.4
Keras-Applications==1.0.8
Keras-Preprocessing==1.1.0
numpy==1.17.0
Pillow==6.1.0
python-dateutil==2.8.0
pytorch-pretrained-bert==0.6.2
PyYAML==5.1.1
regex==2019.6.8
requests==2.22.0
s3transfer==0.2.1
scipy==1.3.0
seqeval==0.0.12
six==1.12.0
torch==1.1.0
torchvision==0.3.0
tqdm==4.32.2
urllib3==1.25.3
导入“ warmup_linear”需要做什么?
答案 0 :(得分:3)
0.4.0 版没有这个问题。
pip install pytorch_pretrained_bert==0.4.0
或者降级到0.6.1解决了。
答案 1 :(得分:1)
将导入行更改为:
from pytorch_pretrained_bert.optimization import BertAdam, WarmupLinearSchedule
在warmup_linear
脚本中没有名为optimization.py
的类。