Marian.md

November 11, 2022 ยท View on GitHub

Marian

You can fine-tune a Marian model from HuggingFace through model=Marian, model_path=<hf-or-local-path>, dataset=<dataset-name>. Specifically, Marian is a Multilingual Model. You can set source language and target language with src_lang=<source language>, tgt_lang=<target_language>.

Example usage:

python run_textbox.py \
    --model=Marian \
    --model_name=marian \
    --model_path=Helsinki-NLP/opus-mt-ROMANCE-en \
    --dataset=wmt16-ro-en \
    --src_lang=ro \
    --tgt_lang=en