Instructions to use prajjwal1/bert-small-mnli with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use prajjwal1/bert-small-mnli with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="prajjwal1/bert-small-mnli")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("prajjwal1/bert-small-mnli") model = AutoModelForSequenceClassification.from_pretrained("prajjwal1/bert-small-mnli", device_map="auto") - Notebooks
- Google Colab
- Kaggle
|
Download README.md from prajjwal1/bert-small-mnli: direct link, hf CLI and curl.
- Browser
- Download file 994 Bytes
-
https://huggingface.co/prajjwal1/bert-small-mnli/resolve/main/README.md
- Command line
-
hf download hf://prajjwal1/bert-small-mnli/README.md
-
curl -L -o README.md https://huggingface.co/prajjwal1/bert-small-mnli/resolve/main/README.md
994 Bytes
| The following model is a Pytorch pre-trained model obtained from converting Tensorflow checkpoint found in the [official Google BERT repository](https://github.com/google-research/bert). These BERT variants were introduced in the paper [Well-Read Students Learn Better: On the Importance of Pre-training Compact Models](https://arxiv.org/abs/1908.08962). These models are trained on MNLI. | |
| If you use the model, please consider citing the paper | |
| ``` | |
| @misc{bhargava2021generalization, | |
| title={Generalization in NLI: Ways (Not) To Go Beyond Simple Heuristics}, | |
| author={Prajjwal Bhargava and Aleksandr Drozd and Anna Rogers}, | |
| year={2021}, | |
| eprint={2110.01518}, | |
| archivePrefix={arXiv}, | |
| primaryClass={cs.CL} | |
| } | |
| ``` | |
| Original Implementation and more info can be found in [this Github repository](https://github.com/prajjwal1/generalize_lm_nli). | |
| ``` | |
| MNLI: 72.1% | |
| MNLI-mm: 73.76% | |
| ``` | |
| These models were trained for 4 epochs. | |
| [@prajjwal_1](https://twitter.com/prajjwal_1) | |