项目作者: RyanDsilva

项目描述 :
Finetuning the BART model on AMI Meetings Corpus
高级语言: Python
项目地址: git://github.com/RyanDsilva/bart-finetuning-ami.git
创建时间: 2020-04-14T11:08:04Z
项目社区:https://github.com/RyanDsilva/bart-finetuning-ami

开源协议:MIT License

下载


BART Finetuning on AMI Meetings Corpus

Setup 💻

  1. git clone https://github.com/RyanDsilva/bart-finetuning-ami.git
  2. pip install -r requirements.txt
  3. python train.py \
  4. --data_dir=./data \
  5. --model_type=bart \
  6. --model_name_or_path=bart-large \
  7. --learning_rate=3e-5 \
  8. --train_batch_size=4 \
  9. --eval_batch_size=4 \
  10. --output_dir=./outputs \
  11. --do_train