【发布时间】:2021-07-16 02:17:15
【问题描述】:
我想将 AllenNLP Interpret(代码 + 演示)与使用 HuggingFace(electra 基鉴别器)训练的 PyTorch 分类模型结合使用。然而,对我来说,如何转换我的模型并在本地 allen-nlp 演示服务器中使用它并不明显。
我应该如何进行?
提前致谢
【问题讨论】:
标签: nlp huggingface-transformers allennlp
我想将 AllenNLP Interpret(代码 + 演示)与使用 HuggingFace(electra 基鉴别器)训练的 PyTorch 分类模型结合使用。然而,对我来说,如何转换我的模型并在本地 allen-nlp 演示服务器中使用它并不明显。
我应该如何进行?
提前致谢
【问题讨论】:
标签: nlp huggingface-transformers allennlp
如果您的任务是二进制分类,您可以查看https://github.com/allenai/allennlp-models/blob/main/training_config/classification/boolq_roberta.jsonnet 中的 BoolQ 示例。您可以更改该配置以使用不同的模型(例如 Electra)。
我们还为解释功能发布了一些新文档:https://guide.allennlp.org/interpret
为了给你一个更具体的答案,我需要了解更多细节,比如你要解决的任务是什么,你是如何训练原始模型的,等等。
【讨论】: