下载havingface预先训练的模型

Question

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

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

Answer 1

安装您的Google驱动器：

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