Views
No views yet
0.8.0.dev01# train w/ shisa-ai/shisa-v1-athenev2-reannotated-filtered
2
3base_model: google/gemma-3-27b-it
4
5load_in_8bit: false
6load_in_4bit: false
7strict: false
8
9datasets:
10 - path: shisa-ai/shisa-v2-best-of-n-athenev2-tulu70b-llama33-only-no-sysprompt
11 type: chat_template
12 field_messages: conversations
13 message_field_role: from
14 message_field_content: value
15 - path: shisa-ai/shisa-v2-roleplaying-sft
16 type: chat_template
17 field_messages: conversations
18 message_property_mappings:
19 role: role
20 content: content
21 roles:
22 system:
23 - system
24 assistant:
25 - gpt
26 - model
27 - assistant
28 user:
29 - human
30 - user
31 roles_to_train: ["assistant"]
32 - path: shisa-ai/translation_set_april_6
33 split: train[:25%]
34 type: chat_template
35 field_messages: conversations
36 message_property_mappings:
37 role: role
38 content: content
39 roles:
40 system:
41 - system
42 assistant:
43 - gpt
44 - model
45 - assistant
46 user:
47 - human
48 - user
49 roles_to_train: ["assistant"]
50 - path: shisa-ai/rewild-set-deepseek-subset
51 split: train[:25%]
52 type: chat_template
53 field_messages: conversations
54 message_property_mappings:
55 role: role
56 content: content
57 roles:
58 system:
59 - system
60 assistant:
61 - gpt
62 - model
63 - assistant
64 user:
65 - human
66 - user
67 roles_to_train: ["assistant"]
68 - path: shisa-ai/magpie-ultra-set
69 split: train[:8%]
70 type: chat_template
71 field_messages: conversations
72 message_property_mappings:
73 role: role
74 content: content
75 roles:
76 system:
77 - system
78 assistant:
79 - gpt
80 - model
81 - assistant
82 user:
83 - human
84 - user
85 roles_to_train: ["assistant"]
86 - path: shisa-ai/magpie-advanced-questions-set
87 split: train[:8%]
88 type: chat_template
89 field_messages: conversations
90 message_property_mappings:
91 role: role
92 content: content
93 roles:
94 system:
95 - system
96 assistant:
97 - gpt
98 - model
99 - assistant
100 user:
101 - human
102 - user
103 roles_to_train: ["assistant"]
104 - path: shisa-ai/japan-magpie-set
105 split: train
106 type: chat_template
107 field_messages: conversations
108 message_property_mappings:
109 role: role
110 content: content
111 roles:
112 system:
113 - system
114 assistant:
115 - gpt
116 - model
117 - assistant
118 user:
119 - human
120 - user
121 roles_to_train: ["assistant"]
122 - path: shisa-ai/shisa-v2-instruction-following-sft
123 split: train[:50%]
124 type: chat_template
125 field_messages: conversations
126 message_property_mappings:
127 role: role
128 content: content
129 roles:
130 system:
131 - system
132 assistant:
133 - gpt
134 - model
135 - assistant
136 user:
137 - human
138 - user
139 roles_to_train: ["assistant"]
140
141dataset_prepared_path: last_run_prepared
142val_set_size: 0.05
143output_dir: ./outputs/ablation-196-finalsft2-shisa-v2-gemma3-27b
144
145sequence_len: 8192
146sample_packing: true
147pad_to_sequence_len: true
148
149# marginal difference
150neftune_noise_alpha: 5
151
152use_wandb: true
153wandb_project: shisa-v2
154wandb_entity: augmxnt
155wandb_name: ablation-196-finalsft2-shisa-v2-gemma3-27b
156
157gradient_accumulation_steps: 2
158micro_batch_size: 1
159num_epochs: 3
160optimizer: paged_adamw_8bit
161lr_scheduler: linear
162learning_rate: 5.4e-6
163
164train_on_inputs: false
165group_by_length: false
166bf16: auto
167fp16:
168tf32: false
169
170gradient_checkpointing: true
171gradient_checkpointing_kwargs:
172 use_reentrant: false
173early_stopping_patience:
174resume_from_checkpoint:
175logging_steps: 1
176xformers_attention:
177flash_attention: true
178
179warmup_steps: 100
180evals_per_epoch: 2
181eval_table_size:
182saves_per_epoch: 0
183save_total_limit: 1 # Only store a single checkpoint
184debug:
185deepspeed: zero3_bf16.json
186weight_decay: 1e-4
187fsdp:
188fsdp_config:
189special_tokens:
190| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 11.3287 | 0.0027 | 1 | 5.4644 |
| 1.1409 | 0.4993 | 182 | 0.5644 |
| 1.0588 | 0.9986 | 364 | 0.5344 |
| 0.9207 | 1.4966 | 546 | 0.5322 |
| 0.8979 | 1.9959 | 728 | 0.5245 |
| 0.7673 | 2.4938 | 910 | 0.5432 |
| 0.7521 | 2.9931 | 1092 | 0.5417 |