https://github.com/ryandsilva/bart-finetuning-ami
Finetuning the BART model on AMI Meetings Corpus
https://github.com/ryandsilva/bart-finetuning-ami
ami-meetings bart nlp summarization transformer
Last synced: about 1 year ago
JSON representation
Finetuning the BART model on AMI Meetings Corpus
- Host: GitHub
- URL: https://github.com/ryandsilva/bart-finetuning-ami
- Owner: RyanDsilva
- License: mit
- Created: 2020-04-14T11:08:04.000Z (about 6 years ago)
- Default Branch: master
- Last Pushed: 2020-04-14T17:58:10.000Z (about 6 years ago)
- Last Synced: 2025-02-03T13:25:02.694Z (over 1 year ago)
- Topics: ami-meetings, bart, nlp, summarization, transformer
- Language: Python
- Size: 416 KB
- Stars: 3
- Watchers: 2
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# BART Finetuning on AMI Meetings Corpus
### Setup 💻
```bash
git clone https://github.com/RyanDsilva/bart-finetuning-ami.git
pip install -r requirements.txt
python train.py \
--data_dir=./data \
--model_type=bart \
--model_name_or_path=bart-large \
--learning_rate=3e-5 \
--train_batch_size=4 \
--eval_batch_size=4 \
--output_dir=./outputs \
--do_train
```