下载拥抱脸预训练模型

Question

一旦我在一本Colab笔记本上下载了一个预先训练过的模型，它就会在我重置笔记本变量之后消失。有什么方法我可以下载模型第二次使用它吗？

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

Answer 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/')