我尝试使用pytorch中的BertModel类加载预先训练好的模型。
我在torch下有_six.py,但它仍然显示模块'torch‘没有属性'_six’。
import torch
from pytorch_pretrained_bert import BertTokenizer, BertModel, BertForMaskedLM
# Load pre-trained model (weights)
model = BertModel.from_pretrained('bert-base-uncased')
model.eval()
~/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py in __setattr__(self, name, value)
551 .format(torch.typename(value), name))
552 modules[name] = value
--> 553 else:
554 buffers = self.__dict__.get('_buffers')
555 if buffers is not None and name in buffers:
~/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py in register_parameter(self, name, param)
140 raise KeyError("parameter name can't be empty string \"\"")
141 elif hasattr(self, name) and name not in self._parameters:
--> 142 raise KeyError("attribute '{}' already exists".format(name))
143
144 if param is None:
AttributeError: module 'torch' has no attribute '_six'
发布于 2019-08-29 17:34:47
在jupyter笔记本中,只需重新启动内核就可以正常工作。
发布于 2019-06-06 12:57:02
我在macOS上也遇到了同样的问题,正如马克提到的,在重启我的mac之后,它工作得很好!
https://stackoverflow.com/questions/56241856
复制相似问题