ailab-prompt-transfer / TextBox

Implement of PTG
https://github.com/RUCAIBox/TextBox
MIT License
0 stars 0 forks source link

BART fine-tuning #17

Open minji-o-j opened 1 year ago

minji-o-j commented 1 year ago

best checkpoint기준 실험 (멈춘것 -3) bart finetuning시 batch 또 조정해줘야함

QKV 학습 버전 (tag: QKVo_#num)


minji-o-j commented 1 year ago

QKV 학습X 버전 (tag: QKVx_#num)



minji-o-j commented 1 year ago

8번 prompt1 말고 accelerate launch run_textbox.py --model=PTG --dataset=xsum --model_path=facebook/bart-large --gpu_id=0,1,2,3 --find_unused_parameters=true --source_task=cross_dataset2_paper --training_option=BART-finetuning --QKV_training=True --learning_rate=3e-5 --attention_path=PTG-xsum-2023-Aug-14_03-53-29_QKVo_8/checkpoint_epoch-4 --train_batch_size=2 --accumulation_steps=48 --eval_batch_size=8