Add adapter distilbert-base-uncased_sentiment_imdb_pfeiffer version 1
Browse files- README.md +57 -0
- adapter_config.json +41 -0
- head_config.json +21 -0
- pytorch_adapter.bin +3 -0
- pytorch_model_head.bin +3 -0
README.md
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
tags:
|
3 |
+
- text-classification
|
4 |
+
- adapter-transformers
|
5 |
+
- distilbert
|
6 |
+
- adapterhub:sentiment/imdb
|
7 |
+
datasets:
|
8 |
+
- imdb
|
9 |
+
license: "apache-2.0"
|
10 |
+
---
|
11 |
+
|
12 |
+
# Adapter `distilbert-base-uncased_sentiment_imdb_pfeiffer` for distilbert-base-uncased
|
13 |
+
|
14 |
+
Adapter for distilbert-base-uncased in Pfeiffer architecture trained on the IMDB dataset for 15 epochs with early stopping and a learning rate of 1e-4.
|
15 |
+
|
16 |
+
|
17 |
+
**This adapter was created for usage with the [Adapters](https://github.com/Adapter-Hub/adapters) library.**
|
18 |
+
|
19 |
+
## Usage
|
20 |
+
|
21 |
+
First, install `adapters`:
|
22 |
+
|
23 |
+
```
|
24 |
+
pip install -U adapters
|
25 |
+
```
|
26 |
+
|
27 |
+
Now, the adapter can be loaded and activated like this:
|
28 |
+
|
29 |
+
```python
|
30 |
+
from adapters import AutoAdapterModel
|
31 |
+
|
32 |
+
model = AutoAdapterModel.from_pretrained("distilbert-base-uncased")
|
33 |
+
adapter_name = model.load_adapter("AdapterHub/distilbert-base-uncased_sentiment_imdb_pfeiffer")
|
34 |
+
model.set_active_adapters(adapter_name)
|
35 |
+
```
|
36 |
+
|
37 |
+
## Architecture & Training
|
38 |
+
|
39 |
+
- Adapter architecture: pfeiffer
|
40 |
+
- Prediction head: classification
|
41 |
+
- Dataset: [IMDb](http://ai.stanford.edu/~amaas/data/sentiment/)
|
42 |
+
|
43 |
+
## Author Information
|
44 |
+
|
45 |
+
- Author name(s): Clifton Poth
|
46 |
+
- Author email: [email protected]
|
47 |
+
- Author links: [Website](https://calpt.github.io), [GitHub](https://github.com/calpt), [Twitter](https://twitter.com/@clifapt)
|
48 |
+
|
49 |
+
|
50 |
+
|
51 |
+
## Citation
|
52 |
+
|
53 |
+
```bibtex
|
54 |
+
|
55 |
+
```
|
56 |
+
|
57 |
+
*This adapter has been auto-imported from https://github.com/Adapter-Hub/Hub/blob/master/adapters/ukp/distilbert-base-uncased_sentiment_imdb_pfeiffer.yaml*.
|
adapter_config.json
ADDED
@@ -0,0 +1,41 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"config": {
|
3 |
+
"adapter_residual_before_ln": false,
|
4 |
+
"cross_adapter": false,
|
5 |
+
"dropout": 0.0,
|
6 |
+
"factorized_phm_W": true,
|
7 |
+
"factorized_phm_rule": false,
|
8 |
+
"hypercomplex_nonlinearity": "glorot-uniform",
|
9 |
+
"init_weights": "bert",
|
10 |
+
"inv_adapter": null,
|
11 |
+
"inv_adapter_reduction_factor": null,
|
12 |
+
"is_parallel": false,
|
13 |
+
"learn_phm": true,
|
14 |
+
"leave_out": [],
|
15 |
+
"ln_after": false,
|
16 |
+
"ln_before": false,
|
17 |
+
"mh_adapter": true,
|
18 |
+
"non_linearity": "swish",
|
19 |
+
"original_ln_after": true,
|
20 |
+
"original_ln_before": false,
|
21 |
+
"output_adapter": true,
|
22 |
+
"phm_bias": true,
|
23 |
+
"phm_c_init": "normal",
|
24 |
+
"phm_dim": 4,
|
25 |
+
"phm_init_range": 0.0001,
|
26 |
+
"phm_layer": false,
|
27 |
+
"phm_rank": 1,
|
28 |
+
"reduction_factor": 16,
|
29 |
+
"residual_before_ln": true,
|
30 |
+
"scaling": 1.0,
|
31 |
+
"shared_W_phm": false,
|
32 |
+
"shared_phm_rule": true,
|
33 |
+
"use_gating": false
|
34 |
+
},
|
35 |
+
"hidden_size": 768,
|
36 |
+
"model_class": "DistilBertAdapterModel",
|
37 |
+
"model_name": "distilbert-base-uncased",
|
38 |
+
"model_type": "distilbert",
|
39 |
+
"name": "imdb",
|
40 |
+
"version": "0.2.0"
|
41 |
+
}
|
head_config.json
ADDED
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"config": {
|
3 |
+
"activation_function": "tanh",
|
4 |
+
"bias": true,
|
5 |
+
"dropout_prob": null,
|
6 |
+
"head_type": "classification",
|
7 |
+
"label2id": {
|
8 |
+
"LABEL_0": 0,
|
9 |
+
"LABEL_1": 1
|
10 |
+
},
|
11 |
+
"layers": 2,
|
12 |
+
"num_labels": 2,
|
13 |
+
"use_pooler": false
|
14 |
+
},
|
15 |
+
"hidden_size": 768,
|
16 |
+
"model_class": "DistilBertAdapterModel",
|
17 |
+
"model_name": "distilbert-base-uncased",
|
18 |
+
"model_type": "distilbert",
|
19 |
+
"name": "imdb",
|
20 |
+
"version": "0.2.0"
|
21 |
+
}
|
pytorch_adapter.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8f298b56b88238dd8f5d3d7fa599579ddd3e840c6a271da72cfa437bd99980ac
|
3 |
+
size 3595686
|
pytorch_model_head.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d2d0ffb2286ae1b9842bb6a80fbb72a398b1ec7add727c114b2819c493b82603
|
3 |
+
size 2370600
|