rambodazimi's picture
Create README.md
b6b5395 verified
metadata
license: apache-2.0
tags:
  - generated_from_trainer
datasets:
  - glue
metrics:
  - pearsonr
  - spearmanr
model-index:
  - name: roberta-base-finetuned-LoRA-STSB
    results:
      - task:
          name: Text Classification
          type: text-classification
        dataset:
          name: glue
          type: glue
          args: stsb
        metrics:
          - name: Pearsonr
            type: pearsonr
            value: 0.9017830359264423
          - name: Spearmanr
            type: spearmanr
            value: 0.899527153781539

roberta-base-finetuned-lora-stsb

This model is a fine-tuned version of roberta-base on the glue dataset. It achieves the following results on the evaluation set:

  • Pearsonr: 0.9018
  • Spearmanr: 0.8995
  • trainable model parameters: 1181185
  • all model parameters: 125827586
  • percentage of trainable model parameters: 0.94%

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 5e-04
  • train_batch_size: 32
  • eval_batch_size: 32
  • seed: 42
  • weight_decay: 0.01
  • rank: 16
  • lora_alpha: 32
  • lora_dropout: 0.05
  • num_epochs: 5