下载 huggingface 预训练模型 [关闭]

【问题标题】：Downloading huggingface pre-trained models [closed]下载 huggingface 预训练模型 [关闭]
【发布时间】：2023-03-10 05:37:01
【问题描述】：

我在 Colab Notebook 上下载了预训练模型后，它会在我重置 notebook 变量后消失。有没有办法可以下载模型以供第二次使用？

tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')

【问题讨论】：

标签： python nlp google-colaboratory huggingface-transformers

【解决方案1】：

安装你的谷歌驱动器：

from google.colab import drive
drive.mount('/content/drive')

做你的事情并保存你的模型：

from transformers import BertTokenizer

tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
tokenizer.save_pretrained('/content/drive/My Drive/tokenizer/')

在新会话中重新加载它：

tokenizer2 = BertTokenizer.from_pretrained('/content/drive/My Drive/tokenizer/')

【讨论】：