DistilBART.md

November 11, 2022 ยท View on GitHub

DistilBART

You can fine-tune a DistilBART model from HuggingFace through model=BART, model_path=<hf-or-local-path>, dataset=<dataset-name>.

Example usage:

python run_textbox.py \
    --model=BART \
    --model_path=sshleifer/distilbart-cnn-12-6 \
    --dataset=samsum