Hugging Face's logo

AdapterHub
/
facebook-bart-base_sts_sts-b_houlsby

Text Classification
Adapters
adapterhub:sts/sts-b
bart
Model card Files Community
facebook-bart-base_sts_sts-b_houlsby / README.md
calpt's picture
calpt
Add adapter facebook-bart-base_sts_sts-b_houlsby version 1
bc0c84c verified
preview code
|
raw
history blame contribute delete
1.36 kB
metadata
tags:
  - text-classification
  - adapter-transformers
  - adapterhub:sts/sts-b
  - bart
license: apache-2.0

Adapter facebook-bart-base_sts_sts-b_houlsby for facebook/bart-base

Adapter for bart-base in Houlsby architecture trained on the STS-B dataset for 15 epochs with early stopping and a learning rate of 1e-4.

This adapter was created for usage with the Adapters library.

Usage

First, install adapters:

pip install -U adapters

Now, the adapter can be loaded and activated like this:

from adapters import AutoAdapterModel

model = AutoAdapterModel.from_pretrained("facebook/bart-base")
adapter_name = model.load_adapter("AdapterHub/facebook-bart-base_sts_sts-b_houlsby")
model.set_active_adapters(adapter_name)

Architecture & Training

  • Adapter architecture: houlsby
  • Prediction head: classification
  • Dataset: STS-B

Author Information

Citation

This adapter has been auto-imported from https://github.com/Adapter-Hub/Hub/blob/master/adapters/ukp/facebook-bart-base_sts_sts-b_houlsby.yaml.