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

8 lines
183 B
YAML
Raw Normal View History

model_name: /home/toolkit/trl/results/c99dab82e83e9c00d8a8618014d64ed5/code/results/final_model
split: test
use_peft: False
load_in_8bit: False
batch_size: 8
fp16: True
sample: False