Files
pythia410m-sft-tldr/code/configs/rouge_sft_pythia410m_trainall_3epochs.yml

7 lines
170 B
YAML
Raw Normal View History

model_name: /home/toolkit/trl/results/bdb07860945be37257a211c5f83a9907/code/results/final_model
split: test
use_peft: False
load_in_8bit: False
batch_size: 32
fp16: True