CDKA / conf /model /llama.yaml
rainstonee's picture
Upload 33 files
c5d3e8d verified
Raw
History Blame Contribute Delete
202 Bytes
name: llama/llama-2-7b-hf
type: CausalLM
epochs: 1
per_device_batch_size: 1
real_batch_size: 32
bf16: True
eval_epochs: 1
early_stopping_patience: 3
max_length: 1024
logging_steps: 1
learning_rate: 2e-4