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

8 lines
182 B
YAML
Raw Normal View History

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