Skip to content

Latest commit

 

History

History
19 lines (16 loc) · 531 Bytes

Adapter.md

File metadata and controls

19 lines (16 loc) · 531 Bytes
Raw

Adapter

You can use Adapter module (though only for BART, T5, and GPT-2) through setting efficient_methods and efficient_kwargs.

For example, you can use Adapter in BART following below commands:

python run_textbox.py \
    --model=BART \
    --model_path=facebook/bart-base \
    --dataset=samsum \
    --efficient_methods=\[\'adapter\'\] \
    --efficient_kwargs=\{\'adapter_mid_dim\':\ 64\}

or yaml equivalent:

efficient_methods: ['adapter']
efficient_kwargs: {'adapter_mid_dim': 64}