allenai / allenai/scibert

How to load finetuned NER model in allenlp?

Abierto
#110 3 comentarios 0 reacciones 0 asignados Ver en GitHub
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.

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.