llama3_full_sft.yaml 963 Bytes
Newer Older
chenych's avatar
chenych committed
1
2
### model
model_name_or_path: meta-llama/Meta-Llama-3-8B-Instruct
luopl's avatar
luopl committed
3
trust_remote_code: true
chenych's avatar
chenych committed
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

### method
stage: sft
do_train: true
finetuning_type: full
use_badam: true
badam_mode: layer
badam_switch_mode: ascending
badam_switch_interval: 50
badam_verbose: 2
# deepspeed: examples/deepspeed/ds_z3_config.json

### dataset
dataset: identity,alpaca_en_demo
template: llama3
luopl's avatar
luopl committed
19
cutoff_len: 2048
chenych's avatar
chenych committed
20
21
22
max_samples: 1000
overwrite_cache: true
preprocessing_num_workers: 16
chenych's avatar
chenych committed
23
dataloader_num_workers: 4
chenych's avatar
chenych committed
24
25
26
27
28
29
30

### output
output_dir: saves/llama3-8b/full/sft
logging_steps: 10
save_steps: 500
plot_loss: true
overwrite_output_dir: true
chenych's avatar
chenych committed
31
32
save_only_model: false
report_to: none  # choices: [none, wandb, tensorboard, swanlab, mlflow]
chenych's avatar
chenych committed
33
34
35
36
37
38
39
40
41
42

### train
per_device_train_batch_size: 1
gradient_accumulation_steps: 8
learning_rate: 1.0e-5
num_train_epochs: 3.0
lr_scheduler_type: cosine
warmup_ratio: 0.1

### eval
chenych's avatar
chenych committed
43
44
45
46
# val_size: 0.1
# per_device_eval_batch_size: 1
# eval_strategy: steps
# eval_steps: 500