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

7 lines
144 B
YAML
Raw Normal View History

model_name: /home/toolkit/huggingface/tldr_sft_pythia7b_trainall_4V100
split: test
use_peft: False
load_in_8bit: False
batch_size: 8
fp16: True