How to load finetuned NER model in allenlp?
- Lenguaje dominante
- Python
- Estrellas
- 1.7k
- Forks
- 232
- Métricas de merge de PR
- Sin PR fusionados en 30 d
Descripción
I have trained the NER model on sciie dataset using the following config:
```
DATASET='sciie'
TASK='ner'
with_finetuning='_finetune' #'_finetune' # or '' for not fine tuning
dataset_size=38124
export BERT_VOCAB=/home/tomaz/neo4j/scibert/model/vocab.txt
export BERT_WEIGHTS=/home/tomaz/neo4j/scibert/model/weights.tar.gz
```
This worked nicely and I got a model.tar.gz as an output. Now when I try to load it in AllenNLP lib:
```
from allennlp.predictors.predictor import Predictor
predictor = Predictor.from_path("model.tar.gz")
```
I get the following error:
> ConfigurationError: bert-pretrained not in acceptable choices for dataset_reader.token_indexers.bert.type: ['single_id', 'characters', 'elmo_characters', 'spacy', 'pretrained_transformer', 'pretrained_transformer_mismatched']. You should either use the --include-package flag to make sure the correct module is loaded, or use a fully qualified class name in your config file like {"model": "my_module.models.MyModel"} to have it imported automatically.
Any idea how to fix this?
Guía de contribución
No hay ninguna guía de contribución indexada para este repositorio
Evaluación
Este issue todavía no se ha evaluado.