File size: 310 Bytes
1904ee8 |
1 2 3 4 5 6 7 8 9 10 11 12 |
output_dir: /home/toolkit/huggingface/openai_summarize_tldr_rbaseline
train_split: train
eval_split: valid[:2000]
###
model_name: mnoukhov/pythia410m-tldr-sft-rm-adapter
new_column_name: reward_baseline
dataset_name: CarperAI/openai_summarize_tldr
load_in_8bit: False
fp16: True
batch_size: 32
max_length: 560
|