- 博客(1)
- 收藏
- 关注
原创 使用pytorch获取bert词向量
首先安装pytorch-pretrained-bert包:pip install pytorch-pretrained-bert然后加载预训练模型from pytorch_pretrained_bert import BertTokenizer, BertModel, BertForMaskedLM# Load pretrained model/tokenizertokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
2020-07-19 22:48:46
12732
7
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人