Views
No views yet

<|im_start|>system
You are Dolphin, a helpful AI assistant.<|im_end|>
<|im_start|>user
{prompt}<|im_end|>
<|im_start|>assistant

0.4.01base_model: 01-ai/Yi-1.5-34B
2model_type: LlamaForCausalLM
3tokenizer_type: LlamaTokenizer
4trust_remote_code: true
5
6# load_in_8bit: false
7# load_in_4bit: true
8# strict: false
9
10# adapter: qlora
11# lora_modules_to_save: [embed_tokens, lm_head]
12
13# lora_r: 32
14# lora_alpha: 16
15# lora_dropout: 0.05
16# lora_target_linear: True
17# lora_fan_in_fan_out:
18
19datasets:
20 - path: /workspace/datasets/dolphin-2.9/dolphin201-sharegpt2.jsonl
21 type: sharegpt
22 conversation: chatml
23 - path: /workspace/datasets/dolphin-2.9/dolphin-coder-translate-sharegpt2.jsonl
24 type: sharegpt
25 conversation: chatml
26 - path: /workspace/datasets/dolphin-2.9/dolphin-coder-codegen-sharegpt2.jsonl
27 type: sharegpt
28 conversation: chatml
29 - path: /workspace/datasets/dolphin-2.9/m-a-p_Code-Feedback-sharegpt-unfiltered.jsonl
30 type: sharegpt
31 conversation: chatml
32 - path: /workspace/datasets/dolphin-2.9/m-a-p_CodeFeedback-Filtered-Instruction-sharegpt-unfiltered.jsonl
33 type: sharegpt
34 conversation: chatml
35 - path: /workspace/datasets/dolphin-2.9/not_samantha_norefusals.jsonl
36 type: sharegpt
37 conversation: chatml
38 - path: /workspace/datasets/dolphin-2.9/Orca-Math-resort-unfiltered.jsonl
39 type: sharegpt
40 conversation: chatml
41 - path: /workspace/datasets/dolphin-2.9/agent_instruct_react_unfiltered.jsonl
42 type: sharegpt
43 conversation: chatml
44 - path: /workspace/datasets/dolphin-2.9/toolbench_instruct_j1s1_3k_unfiltered.jsonl
45 type: sharegpt
46 conversation: chatml
47 - path: /workspace/datasets/dolphin-2.9/toolbench_negative_unfiltered.jsonl
48 type: sharegpt
49 conversation: chatml
50 - path: /workspace/datasets/dolphin-2.9/toolbench_react_10p_unfiltered.jsonl
51 type: sharegpt
52 conversation: chatml
53 - path: /workspace/datasets/dolphin-2.9/toolbench_tflan_cot_30p_unfiltered.jsonl
54 type: sharegpt
55 conversation: chatml
56 - path: /workspace/datasets/dolphin-2.9/openhermes200k_unfiltered.jsonl
57 type: sharegpt
58 conversation: chatml
59
60chat_template: chatml
61
62dataset_prepared_path: yi34b
63val_set_size: 0.01
64output_dir: ./out-yi
65
66sequence_len: 8192
67sample_packing: true
68pad_to_sequence_len: true
69
70wandb_project: dolphin-2.9-yi-34b
71wandb_watch:
72wandb_run_id:
73wandb_log_model:
74
75gradient_accumulation_steps: 8
76micro_batch_size: 1
77num_epochs: 3
78optimizer: adamw_8bit
79lr_scheduler: cosine
80learning_rate: 1e-5
81
82train_on_inputs: false
83group_by_length: false
84bf16: auto
85fp16:
86tf32: true
87
88gradient_checkpointing: true
89gradient_checkpointing_kwargs:
90 use_reentrant: false
91early_stopping_patience:
92# resume_from_checkpoint: /workspace/axolotl/dbrx-checkpoint
93logging_steps: 1
94xformers_attention:
95flash_attention: true
96
97warmup_steps: 10
98evals_per_epoch: 4
99eval_table_size:
100saves_per_epoch: 4
101save_total_limit: 2
102save_steps:
103debug:
104deepspeed: /workspace/axolotl/deepspeed_configs/zero3_bf16.json
105weight_decay: 0.05
106fsdp:
107fsdp_config:
108special_tokens:
109 bos_token: "<|startoftext|>"
110 eos_token: "<|im_end|>"
111 pad_token: "<unk>"
112 unk_token: "<unk>"
113tokens:
114 - "<|im_start|>"
115
116| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 0.6265 | 0.0 | 1 | 0.6035 |
| 0.4674 | 0.25 | 327 | 0.4344 |
| 0.4337 | 0.5 | 654 | 0.4250 |
| 0.4346 | 0.75 | 981 | 0.4179 |
| 0.3985 | 1.0 | 1308 | 0.4118 |
| 0.3128 | 1.23 | 1635 | 0.4201 |
| 0.3261 | 1.48 | 1962 | 0.4157 |
| 0.3259 | 1.73 | 2289 | 0.4122 |
| 0.3126 | 1.98 | 2616 | 0.4079 |
| 0.2265 | 2.21 | 2943 | 0.4441 |
| 0.2297 | 2.46 | 3270 | 0.4427 |
| 0.2424 | 2.71 | 3597 | 0.4425 |