【发布时间】:2021-08-22 15:20:37
【问题描述】:
我想离线使用 arabert-v2 预处理。但似乎保存模型对于 arabert 预处理功能是不够的。我还应该下载什么?有什么想法吗?
【问题讨论】:
标签: python bert-language-model
我想离线使用 arabert-v2 预处理。但似乎保存模型对于 arabert 预处理功能是不够的。我还应该下载什么?有什么想法吗?
【问题讨论】:
标签: python bert-language-model
一旦您下载了模型和标记器,例如:
model_name = "aubmindlab/aragpt2-mega"
model = GPT2LMHeadModel.from_pretrained(model_name)
tokenizer = GPT2TokenizerFast.from_pretrained(model_name)
然后您可以将它们保存在本地:
model.save_pretrained("../model")
tokenizer.save_pretrained("../tokenizer")
您可以通过以下方式访问模型:
model = GPT2LMHeadModel.from_pretrained("./model")
【讨论】: