Small Domain Models
Collection
Small models (22.7M) trained for specific domains • 8 items • Updated • 1
CeleBERTy Small
This is a 22.7M parameter BERT encoder-only model trained on Wikipedia articles labeled as celebrity related.
This is a domain-specialized small model that often performs as good as models 10-100x larger. It demonstrates that narrowing down a model to a small domain requires less overall parameters than models generalized for all problems.
Usage
celeberty-small can be loaded using Hugging Face Transformers as follows.
from transformers import AutoModel
model = AutoModel.from_pretrained("neuml/celeberty-small")
The model is intended to be further fine-tuned for a specific task such as Text Classification, Entity Extraction, Sentence Embeddings and so on.
More Information
Read more about the model in this article.
- Downloads last month
- 16
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support