urchade / GLiNER

Generalist and Lightweight Model for Named Entity Recognition (Extract any entity types from texts) @ NAACL 2024
https://arxiv.org/abs/2311.08526
Apache License 2.0
1.37k stars 118 forks source link

Clarification of model size in paper #142

Closed SCccc21 closed 4 months ago

SCccc21 commented 4 months ago

Hi, I noticed that in paper the GLiNER s/m/l are of size 50M, 90M, 0.3B respectively, while the published pretrained models have size of 166M, 209M, 459M. Could you please clarify what makes this difference?

SCccc21 commented 4 months ago

Also, why GoLLIE is excluded from the evaluation of the 20 NER dataset in the paper?

urchade commented 4 months ago

Hi @SCccc21 1) All model sizes exclude the embedding layer parameters 2) not sure if in gollie paper, the authors evaluated on all these datasets

Do not hesitate if you have any other questions :)