calpt commited on
Commit
7c34a04
1 Parent(s): 943672f

Initial version.

Browse files
README.md ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - roberta
4
+ - adapterhub:sentiment/imdb
5
+ - adapter-transformers
6
+ datasets:
7
+ - imdb
8
+ language:
9
+ - en
10
+ ---
11
+
12
+ # Adapter `AdapterHub/roberta-base-pf-imdb` for roberta-base
13
+
14
+ An [adapter](https://adapterhub.ml) for the `roberta-base` model that was trained on the [sentiment/imdb](https://adapterhub.ml/explore/sentiment/imdb/) dataset and includes a prediction head for classification.
15
+
16
+ This adapter was created for usage with the **[adapter-transformers](https://github.com/Adapter-Hub/adapter-transformers)** library.
17
+
18
+ ## Usage
19
+
20
+ First, install `adapter-transformers`:
21
+
22
+ ```
23
+ pip install -U adapter-transformers
24
+ ```
25
+ _Note: adapter-transformers is a fork of transformers that acts as a drop-in replacement with adapter support. [More](https://docs.adapterhub.ml/installation.html)_
26
+
27
+ Now, the adapter can be loaded and activated like this:
28
+
29
+ ```python
30
+ from transformers import AutoModelWithHeads
31
+
32
+ model = AutoModelWithHeads.from_pretrained("roberta-base")
33
+ adapter_name = model.load_adapter("AdapterHub/roberta-base-pf-imdb", source="hf")
34
+ model.active_adapters = adapter_name
35
+ ```
36
+
37
+ ## Architecture & Training
38
+
39
+ The training code for this adapter is available at https://github.com/adapter-hub/efficient-task-transfer.
40
+ In particular, training configurations for all tasks can be found [here](https://github.com/adapter-hub/efficient-task-transfer/tree/master/run_configs).
41
+
42
+
43
+ ## Evaluation results
44
+
45
+ Refer to [the paper](https://arxiv.org/pdf/2104.08247) for more information on results.
46
+
47
+ ## Citation
48
+
49
+ If you use this adapter, please cite our paper ["What to Pre-Train on? Efficient Intermediate Task Selection"](https://arxiv.org/pdf/2104.08247):
50
+
51
+ ```bibtex
52
+ @inproceedings{poth-etal-2021-what-to-pre-train-on,
53
+ title={What to Pre-Train on? Efficient Intermediate Task Selection},
54
+ author={Clifton Poth and Jonas Pfeiffer and Andreas Rücklé and Iryna Gurevych},
55
+ booktitle = "Proceedings of the 2021 Conference on Empirical Methods in Natural Language Processing (EMNLP)",
56
+ month = nov,
57
+ year = "2021",
58
+ address = "Online",
59
+ publisher = "Association for Computational Linguistics",
60
+ url = "https://arxiv.org/abs/2104.08247",
61
+ pages = "to appear",
62
+ }
63
+ ```
adapter_config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "config": {
3
+ "adapter_residual_before_ln": false,
4
+ "cross_adapter": false,
5
+ "inv_adapter": null,
6
+ "inv_adapter_reduction_factor": null,
7
+ "leave_out": [],
8
+ "ln_after": false,
9
+ "ln_before": false,
10
+ "mh_adapter": false,
11
+ "non_linearity": "relu",
12
+ "original_ln_after": true,
13
+ "original_ln_before": true,
14
+ "output_adapter": true,
15
+ "reduction_factor": 16,
16
+ "residual_before_ln": true
17
+ },
18
+ "hidden_size": 768,
19
+ "model_class": "RobertaModelWithHeads",
20
+ "model_name": "roberta-base",
21
+ "model_type": "roberta",
22
+ "name": "imdb"
23
+ }
head_config.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "config": {
3
+ "activation_function": "tanh",
4
+ "bias": true,
5
+ "head_type": "classification",
6
+ "label2id": {
7
+ "LABEL_0": 0,
8
+ "LABEL_1": 1
9
+ },
10
+ "layers": 2,
11
+ "num_labels": 2,
12
+ "use_pooler": false
13
+ },
14
+ "hidden_size": 768,
15
+ "model_class": "RobertaModelWithHeads",
16
+ "model_name": "roberta-base",
17
+ "model_type": "roberta",
18
+ "name": "imdb"
19
+ }
pytorch_adapter.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:16cf8f768c5ab62e34f820f984e8ed04224d15f67574a60632f3891b9fa21bc3
3
+ size 3594735
pytorch_model_head.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c01467b077fbcbf6f0406e081ca64b32632ed52fb5fc737ce51327014c1ae781
3
+ size 2370175