Skip to content

Latest commit

 

History

History
16 lines (11 loc) · 579 Bytes

extract_model.md

File metadata and controls

16 lines (11 loc) · 579 Bytes

Extracting a model after finetuning

back to main README

If you would like to re-use your MaChAmp finetuned model in another toolkit, you can extract the transformers model. We provide an example script in scripts/misc/extract_automodel.py, which needs to be ran from the root of this repo. Usage is as follows:

cp scripts/misc/extract_automodel.py .
python3 extract_automodel.py logs/ewt/*/model.pt mBERT_finetuned_on_EWT

Now the models including its configuration and tokenizer will be written in a folder titled mBERT_finetuned_on_EWT.