How load tokenizer and model after training ?

sayef / fsner

Few-shot Named Entity Recognition

119 stars 6 forks source link

How load tokenizer and model after training ? #10

Closed ByUnal closed 2 years ago

ByUnal commented 2 years ago

I have these files after training. How can I load the tokenizer and model for testing or inference ?

sayef commented 2 years ago

I assume these files are inside a folder, let's say "pretrained_model". Now please see the example in the readme and replace "sayef/fsner-bert-base-uncased" with "./pretraimed_model".

ByUnal commented 2 years ago

Worked, thanks.