Description
The baseline NLI model is a fine-tuned version of bert-base-uncased for Text Classifacation on the MNLI dataset , with entailments as label 0 and all others (neutral or contradiction) as label 1.
Two classes:
entailment: 0
non-entailment: 1
Results on the validation set
MNLI: 90.07%
MNLI-mm: 90.08%
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 2e-05
- train_batch_size: 32
- eval_batch_size: 32
- optimizer: "adamw_torch"
- lr_scheduler_type: linear
- num_epochs: 4
- Downloads last month
- 9
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social
visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.