alwaysgood commited on
Commit
b9f8827
·
verified ·
1 Parent(s): 8efacd8

Upload DQS DPO final evaluation results

Browse files
eval/final/.ipynb_checkpoints/eval_outputs-checkpoint.jsonl ADDED
The diff for this file is too large to render. See raw diff
 
eval/final/effective_config.yaml ADDED
@@ -0,0 +1,422 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ run:
2
+ seed: 42
3
+ student_filter_tag: sf_on
4
+ id: gemma4_e2b_dqs_dpo_full_response_seed42
5
+ profile: real
6
+ subset_start: 0
7
+ subset_end: 23
8
+ hydra_chdir: false
9
+ paths:
10
+ artifact_root: artifacts/runs/gemma4_e2b_dqs_dpo_full_response_seed42
11
+ cache_dir: artifacts/cache
12
+ prepared_data_dir: artifacts/prepared
13
+ checkpoint_dir: artifacts/runs/gemma4_e2b_dqs_dpo_full_response_seed42/checkpoints
14
+ subset_dir: artifacts/runs/gemma4_e2b_dqs_dpo_full_response_seed42/subsets
15
+ config_snapshot_path: artifacts/runs/gemma4_e2b_dqs_dpo_full_response_seed42/effective_config.yaml
16
+ config_hash_path: artifacts/runs/gemma4_e2b_dqs_dpo_full_response_seed42/config_hash.txt
17
+ data:
18
+ raw_dataset_repo: alwaysgood/financial-english-source-corpus
19
+ prepared_dataset_repo: alwaysgood/financial-english-source-corpus-gemma4-e2b-1280
20
+ use_prepared_dataset: true
21
+ prepared_download:
22
+ revision: main
23
+ local_dir: data/prepared/financial-english-source-corpus-gemma4-e2b-1280
24
+ workers: 16
25
+ allow_patterns:
26
+ - '*.parquet'
27
+ - '*.json'
28
+ - '*.yaml'
29
+ - '*.yml'
30
+ - '*.txt'
31
+ - README.md
32
+ preprocess_raw:
33
+ split: train
34
+ revision: main
35
+ tokenizer_model: google/gemma-4-E2B-it
36
+ tokenizer_revision: main
37
+ output_dir: data/prepared/financial-english-source-corpus-gemma4-e2b-1280-rebuilt
38
+ workers: 16
39
+ rows_per_shard: 50000
40
+ task_buffer_size: 256
41
+ source_field: source
42
+ target_field: target
43
+ metadata_field: metadata
44
+ max_input_tokens: 1280
45
+ max_output_tokens: 1500
46
+ length_counter: tokenizer
47
+ split_long_sources: false
48
+ subset_size: 100000
49
+ selection_ratio: 0.01
50
+ teacher_target_per_subset: 1000
51
+ qe_selection_order: low
52
+ raw_random_baseline: false
53
+ length_bucket_selection:
54
+ enabled: true
55
+ quota_strategy: proportional
56
+ weights: null
57
+ fill_remainder_from_global: true
58
+ allow_zero_weight_bucket_fill: false
59
+ length_bucket_sampling_profiles:
60
+ financial_3bucket_weighted:
61
+ quota_strategy: weighted
62
+ length_buckets:
63
+ - - 1
64
+ - 64
65
+ - - 65
66
+ - 256
67
+ - - 257
68
+ - 1280
69
+ weights:
70
+ - 0.25
71
+ - 0.65
72
+ - 0.1
73
+ length_buckets:
74
+ - - 1
75
+ - 64
76
+ - - 65
77
+ - 256
78
+ - - 257
79
+ - 1280
80
+ degeneration_filter:
81
+ enabled: true
82
+ student_enabled: true
83
+ teacher_enabled: true
84
+ student_require_valid_output: true
85
+ reject_repetition: true
86
+ reject_thinking_trace: true
87
+ reject_prompt_artifact: true
88
+ reject_foreign_passthrough: false
89
+ reject_english_passthrough: true
90
+ reject_wrong_script_heavy: true
91
+ reject_wrong_script_light: true
92
+ reject_truncation: true
93
+ reject_encoding_replchar: true
94
+ reject_length_explosion: true
95
+ reject_low_hangul_other: true
96
+ foreign_passthrough:
97
+ min_text_chars: 80
98
+ low_hangul_ratio_max: 0.1
99
+ low_hangul_latin_like_ratio_min: 0.5
100
+ hangul_ratio_max: 0.45
101
+ latin_like_ratio_min: 0.3
102
+ source_token_overlap_min: 0.12
103
+ english_passthrough:
104
+ hangul_ratio_max: 0.05
105
+ latin_ratio_min: 0.5
106
+ source_token_overlap_min: 0.55
107
+ partial_copy_min_text_chars: 6
108
+ partial_copy_min_source_chars: 20
109
+ partial_copy_hangul_ratio_max: 0.05
110
+ partial_copy_latin_ratio_min: 0.5
111
+ partial_copy_mt_source_precision_min: 0.8
112
+ partial_copy_max_mt_tokens: 12
113
+ partial_copy_min_source_tokens: 5
114
+ sentence_min_text_chars: 40
115
+ sentence_hangul_ratio_max: 0.05
116
+ sentence_latin_like_ratio_min: 0.35
117
+ sentence_source_token_overlap_min: 0.55
118
+ prompt_artifact:
119
+ labels:
120
+ min_text_chars: 8
121
+ latin_like_ratio_min: 0.2
122
+ hangul_ratio_max: 0.45
123
+ wrong_script_light:
124
+ min_text_chars: 30
125
+ wrong_script_char_count_min: 3
126
+ mixed_script_token_count_min: 999999
127
+ repetition:
128
+ max_analysis_chars: 6000
129
+ max_analysis_tokens: 500
130
+ max_char_unit_len: 16
131
+ max_immediate_span_tokens: 24
132
+ char_repeat_run_min: 10
133
+ char_repeat_excess_over_source_min: 8
134
+ word_repeat_run_min: 8
135
+ word_repeat_excess_over_source_min: 5
136
+ immediate_span:
137
+ token_span_min: 2
138
+ repeat_run_min: 3
139
+ repeated_token_score_min: 8
140
+ excess_score_over_source_min: 6
141
+ long_span_token_span_min: 6
142
+ long_span_repeat_run_min: 2
143
+ long_span_repeated_token_score_min: 14
144
+ long_span_excess_score_over_source_min: 8
145
+ token_ngram:
146
+ n_min: 3
147
+ n_max: 12
148
+ repeat_count_min: 4
149
+ coverage_min: 0.18
150
+ repeated_token_excess_over_source_min: 8
151
+ short_ngram:
152
+ n_min: 2
153
+ n_max: 2
154
+ cheap_token_frequency_min: 8
155
+ repeat_count_min: 8
156
+ coverage_min: 0.04
157
+ repeated_token_excess_over_source_min: 10
158
+ token_frequency:
159
+ count_min: 30
160
+ ratio_min: 0.06
161
+ count_excess_over_source_min: 20
162
+ korean_function_token:
163
+ count_min: 15
164
+ ratio_min: 0.1
165
+ normalized_clause_duplication:
166
+ min_normalized_chars: 10
167
+ duplicate_count_min: 3
168
+ duplicate_ratio_min: 0.3
169
+ duplicate_count_excess_over_source_min: 2
170
+ short_duplicate_count_min: 2
171
+ short_duplicate_ratio_min: 0.8
172
+ short_duplicate_count_excess_over_source_min: 1
173
+ short_text_chars_max: 1000
174
+ truncation:
175
+ min_dangling_text_chars: 30
176
+ min_dangling_particle_text_chars: 80
177
+ dangling_korean_suffixes:
178
+ - 이었으며
179
+ - 였으며
180
+ - 었으며
181
+ - 했으며
182
+ - 되었으며
183
+ - 이며
184
+ - 하며
185
+ - 하고
186
+ - 했고
187
+ - 하였고
188
+ - 되었고
189
+ - 하기 위해
190
+ - 위해
191
+ - 때문에
192
+ dangling_korean_particles:
193
+ - 에
194
+ - 의
195
+ - 로
196
+ - 으로
197
+ - 와
198
+ - 과
199
+ - 은
200
+ - 는
201
+ - 을
202
+ - 를
203
+ - 이
204
+ - 가
205
+ - 에서
206
+ - 부터
207
+ - 까지
208
+ model:
209
+ name_or_path: google/gemma-4-E2B-it
210
+ prepared_dataset_repo: alwaysgood/financial-english-source-corpus-gemma4-e2b-1280
211
+ prepared_dataset_slug: financial-english-source-corpus-gemma4-e2b-1280
212
+ family: gemma4
213
+ size: e2b
214
+ variant: instruct
215
+ is_vision_model: true
216
+ unsloth_model_api: fast_model
217
+ trust_remote_code: false
218
+ use_hf_chat_template: true
219
+ enable_thinking: false
220
+ require_thinking_control: false
221
+ ddp_find_unused_parameters: false
222
+ vllm_limit_mm_per_prompt:
223
+ image: 0
224
+ audio: 0
225
+ tokenizer_revision: main
226
+ prompts:
227
+ template_seed: 42
228
+ student_templates_path: prompts/student_templates.yaml
229
+ teacher_template_path: prompts/teacher_template.txt
230
+ student_selection:
231
+ random_template: true
232
+ deterministic_by_sample_id: true
233
+ use_language_variants: true
234
+ sft_training:
235
+ random_template: true
236
+ deterministic_by_sample_id: true
237
+ use_language_variants: true
238
+ evaluation:
239
+ random_template: false
240
+ base_template_id: base_tem_001
241
+ instruct_template_id: instruct_tem_003
242
+ inference:
243
+ backend: vllm
244
+ temperature: 0.0
245
+ top_p: 1.0
246
+ max_new_tokens: 1500
247
+ num_gpus: 1
248
+ tensor_parallel_size: 1
249
+ data_parallel_replicas: auto
250
+ gpu_ids: null
251
+ shard_strategy: order_split
252
+ gpu_memory_utilization: 0.9
253
+ save_outputs: true
254
+ allow_fallback: false
255
+ qe:
256
+ selection:
257
+ backend: comet
258
+ model: Unbabel/wmt22-cometkiwi-da
259
+ requires_reference: false
260
+ python_env_var: COMET_PYTHON
261
+ batch_size: 512
262
+ num_gpus: 1
263
+ gpu_ids: null
264
+ shard_strategy: order_split
265
+ allow_fallback: false
266
+ eval:
267
+ profile: final
268
+ dataset_path: data/test.jsonl
269
+ output_dir: artifacts/runs/gemma4_e2b_dqs_dpo_full_response_seed42/eval/final
270
+ source_field: source
271
+ target_field: target
272
+ metadata_field: metadata
273
+ limit: null
274
+ generation:
275
+ model_path: auto
276
+ max_new_tokens: 1024
277
+ temperature: 0.0
278
+ top_p: 1.0
279
+ num_gpus: 1
280
+ tensor_parallel_size: 1
281
+ gpu_memory_utilization: 0.9
282
+ filter:
283
+ enabled: true
284
+ degeneration_filter_overrides:
285
+ wrong_script_light:
286
+ wrong_script_char_count_min: 2
287
+ metrics:
288
+ - id: bleu
289
+ backend: sacrebleu
290
+ metric: bleu
291
+ requires_reference: true
292
+ - id: chrf
293
+ backend: sacrebleu
294
+ metric: chrf
295
+ requires_reference: true
296
+ - id: comet_wmt22_da
297
+ backend: comet
298
+ model: Unbabel/wmt22-comet-da
299
+ requires_reference: true
300
+ python_env_var: COMET_PYTHON
301
+ batch_size: 16
302
+ - id: cometkiwi_wmt22_da
303
+ backend: comet
304
+ model: Unbabel/wmt22-cometkiwi-da
305
+ requires_reference: false
306
+ python_env_var: COMET_PYTHON
307
+ batch_size: 16
308
+ - id: xcomet_xxl
309
+ backend: comet
310
+ model: Unbabel/XCOMET-XXL
311
+ requires_reference: true
312
+ python_env_var: COMET_PYTHON
313
+ batch_size: 16
314
+ - id: metricx_24_hybrid_xxl
315
+ backend: metricx
316
+ model: google/metricx-24-hybrid-xxl-v2p6-bfloat16
317
+ tokenizer: google/mt5-xl
318
+ requires_reference: true
319
+ python_env_var: METRICX_PYTHON
320
+ module: metricx24.predict
321
+ max_input_length: 1536
322
+ batch_size: 4
323
+ teacher:
324
+ max_workers: 25
325
+ batch_size: 4
326
+ candidate_multiplier: 4
327
+ refill_until_target: true
328
+ structured_output: true
329
+ max_retries_per_row: 3
330
+ abort_on_all_failed_window: true
331
+ max_output_tokens: 8192
332
+ reject_over_max_output_tokens: true
333
+ target_max_output_tokens: 1500
334
+ temperature: 0.0
335
+ invalid_label: invalid
336
+ prompt_path: prompts/teacher_template.txt
337
+ system_prompt_path: prompts/teacher_system.txt
338
+ draft_format_policy_path: prompts/teacher_draft_policy_strict.txt
339
+ user_prompt_path: prompts/teacher_user_batch.txt
340
+ allow_fallback: false
341
+ providers:
342
+ - name: openai
343
+ model: gpt-4.1
344
+ weight: 0.0
345
+ api_key_env: OPENAI_API_KEY
346
+ - name: anthropic
347
+ model: claude-sonnet-4-20250514
348
+ weight: 0.0
349
+ api_key_env: ANTHROPIC_API_KEY
350
+ - name: gemini
351
+ model: gemini-3.1-flash-lite
352
+ weight: 1.0
353
+ api_key_env: GEMINI_API_KEY
354
+ training:
355
+ backend: unsloth
356
+ tuning_mode: full
357
+ dtype: auto
358
+ load_in_4bit: false
359
+ load_in_8bit: false
360
+ gradient_checkpointing: unsloth
361
+ learning_rate: 2.0e-05
362
+ warmup_ratio: 0.1
363
+ scheduler: cosine
364
+ stage_planned_rows_per_subset: 1000
365
+ optimizer: adamw_torch
366
+ max_grad_norm: 5.0
367
+ weight_decay: 0.0
368
+ num_train_epochs: 1
369
+ max_steps: -1
370
+ save_strategy: steps
371
+ save_steps: 100
372
+ save_total_limit: 1
373
+ logging_steps: 1
374
+ dataloader_num_workers: 4
375
+ dataloader_shuffle: true
376
+ ddp_find_unused_parameters: true
377
+ resume_from_checkpoint: auto
378
+ append_eos_token: true
379
+ response_only_loss: true
380
+ effective_batch_size: 128
381
+ per_device_train_batch_size: 4
382
+ per_device_eval_batch_size: 4
383
+ gradient_accumulation_steps: auto
384
+ packing: false
385
+ lora: null
386
+ train_vision_layers: false
387
+ full_finetuning: true
388
+ optimizer_resume_required: true
389
+ scheduler_resume_required: true
390
+ global_step_resume_required: true
391
+ save_optimizer_state: true
392
+ save_scheduler_state: true
393
+ save_adapter: false
394
+ save_merged_model: false
395
+ merge_smoke_test_required: false
396
+ merge_smoke_test_mode: config_and_tokenizer
397
+ save_final_model: true
398
+ save_full_model: false
399
+ normalize_full_weight_checkpoint_keys: true
400
+ assert_full_weight_checkpoint_keys: true
401
+ max_seq_length: null
402
+ prompt_overhead_tokens: 128
403
+ prevent_template_truncation: true
404
+ logging:
405
+ local:
406
+ enabled: true
407
+ root_dir: artifacts/runs/gemma4_e2b_dqs_dpo_full_response_seed42/logs
408
+ wandb:
409
+ enabled: true
410
+ project: dqs
411
+ entity: null
412
+ run_name: gemma4_e2b_dqs_dpo_full_response_seed42
413
+ run_id: gemma4_e2b_dqs_dpo_full_response_seed42
414
+ resume: allow
415
+ tags:
416
+ - dqs
417
+ - qwen3.5
418
+ - en-ko
419
+ save_effective_config: true
420
+ save_template_versions: true
421
+ save_all_step_artifacts: false
422
+ version: 1
eval/final/eval_outputs.jsonl ADDED
The diff for this file is too large to render. See raw diff
 
eval/final/eval_records.jsonl ADDED
The diff for this file is too large to render. See raw diff
 
eval/final/eval_summary.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset_path": "data/test.jsonl",
3
+ "dry_run": false,
4
+ "eval_profile": "final",
5
+ "filter_fail_ratio": 0.0,
6
+ "filter_fail_rows": 0,
7
+ "filter_label_counts": {
8
+ "clean": 500
9
+ },
10
+ "filter_pass_rows": 500,
11
+ "generation_ok_rows": 500,
12
+ "lora_adapter_path": null,
13
+ "metrics": {
14
+ "bleu": {
15
+ "backend": "sacrebleu",
16
+ "higher_is_better": true,
17
+ "metric": "bleu",
18
+ "row_score": "sentence_bleu",
19
+ "score": 29.36231018505856
20
+ },
21
+ "chrf": {
22
+ "backend": "sacrebleu",
23
+ "higher_is_better": true,
24
+ "metric": "chrf",
25
+ "row_score": "sentence_chrf",
26
+ "score": 49.60279834708574
27
+ },
28
+ "comet_wmt22_da": {
29
+ "backend": "comet",
30
+ "higher_is_better": true,
31
+ "mean": 0.8950758044719695,
32
+ "model": "Unbabel/wmt22-comet-da"
33
+ },
34
+ "cometkiwi_wmt22_da": {
35
+ "backend": "comet",
36
+ "higher_is_better": true,
37
+ "mean": 0.8539881974458694,
38
+ "model": "Unbabel/wmt22-cometkiwi-da"
39
+ },
40
+ "metricx_24_hybrid_xxl": {
41
+ "backend": "metricx",
42
+ "higher_is_better": false,
43
+ "mean": 3.5188203125,
44
+ "model": "google/metricx-24-hybrid-xxl-v2p6-bfloat16"
45
+ },
46
+ "xcomet_xxl": {
47
+ "backend": "comet",
48
+ "higher_is_better": true,
49
+ "mean": 0.8844950605034828,
50
+ "model": "Unbabel/XCOMET-XXL"
51
+ }
52
+ },
53
+ "model_path": "post_training/dqs_preference_training_hf/outputs/gemma4_e2b_dqs_dpo_full_response_seed42/vllm_final",
54
+ "output_dir": "post_training/dqs_preference_training_hf/outputs/gemma4_e2b_dqs_dpo_full_response_seed42/eval/final",
55
+ "rows": 500,
56
+ "run_id": "gemma4_e2b_dqs_dpo_full_response_seed42",
57
+ "usage": {}
58
+ }