Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,28 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+checkpoint-100/math.test.v1.1.0shot.jsonl filter=lfs diff=lfs merge=lfs -text
+checkpoint-100/tokenizer.json filter=lfs diff=lfs merge=lfs -text
+checkpoint-200/math.test.v1.1.0shot.jsonl filter=lfs diff=lfs merge=lfs -text
+checkpoint-200/tokenizer.json filter=lfs diff=lfs merge=lfs -text
+checkpoint-300/math.test.v1.1.0shot.json filter=lfs diff=lfs merge=lfs -text
+checkpoint-300/math.test.v1.1.0shot.jsonl filter=lfs diff=lfs merge=lfs -text
+checkpoint-300/tokenizer.json filter=lfs diff=lfs merge=lfs -text
+checkpoint-400/math.test.v1.1.0shot.json filter=lfs diff=lfs merge=lfs -text
+checkpoint-400/math.test.v1.1.0shot.jsonl filter=lfs diff=lfs merge=lfs -text
+checkpoint-400/tokenizer.json filter=lfs diff=lfs merge=lfs -text
+checkpoint-500/math.test.v1.1.0shot.json filter=lfs diff=lfs merge=lfs -text
+checkpoint-500/math.test.v1.1.0shot.jsonl filter=lfs diff=lfs merge=lfs -text
+checkpoint-500/tokenizer.json filter=lfs diff=lfs merge=lfs -text
+checkpoint-600/math.test.v1.1.0shot.json filter=lfs diff=lfs merge=lfs -text
+checkpoint-600/math.test.v1.1.0shot.jsonl filter=lfs diff=lfs merge=lfs -text
+checkpoint-600/tokenizer.json filter=lfs diff=lfs merge=lfs -text
+checkpoint-700/math.test.v1.1.0shot.json filter=lfs diff=lfs merge=lfs -text
+checkpoint-700/math.test.v1.1.0shot.jsonl filter=lfs diff=lfs merge=lfs -text
+checkpoint-700/tokenizer.json filter=lfs diff=lfs merge=lfs -text
+checkpoint-800/math.test.v1.1.0shot.json filter=lfs diff=lfs merge=lfs -text
+checkpoint-800/math.test.v1.1.0shot.jsonl filter=lfs diff=lfs merge=lfs -text
+checkpoint-800/tokenizer.json filter=lfs diff=lfs merge=lfs -text
+checkpoint-900/math.test.v1.1.0shot.json filter=lfs diff=lfs merge=lfs -text
+checkpoint-900/math.test.v1.1.0shot.jsonl filter=lfs diff=lfs merge=lfs -text
+checkpoint-900/tokenizer.json filter=lfs diff=lfs merge=lfs -text

checkpoint-100/config.json

@@ -0,0 +1,28 @@
+{
+  "_name_or_path": "experiments/gemma.2b.it.meta_math_distil.H100.w4.v1.0/checkpoint-2000/",
+  "architectures": [
+    "GemmaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "bos_token_id": 2,
+  "eos_token_id": 1,
+  "head_dim": 256,
+  "hidden_act": "gelu",
+  "hidden_size": 2048,
+  "initializer_range": 0.02,
+  "intermediate_size": 16384,
+  "max_position_embeddings": 8192,
+  "model_type": "gemma",
+  "num_attention_heads": 8,
+  "num_hidden_layers": 18,
+  "num_key_value_heads": 1,
+  "pad_token_id": 0,
+  "rms_norm_eps": 1e-06,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.38.2",
+  "use_cache": true,
+  "vocab_size": 256000
+}

checkpoint-100/generation_config.json

@@ -0,0 +1,7 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 2,
+  "eos_token_id": 1,
+  "pad_token_id": 0,
+  "transformers_version": "4.38.2"
+}

checkpoint-100/gsm8k.test.v1.1.0shot.metrics.json

@@ -0,0 +1,5 @@
+{
+  "acc": 0.5261561789234268,
+  "correct": 694,
+  "total": 1319
+}

checkpoint-100/math.test.v1.1.0shot.jsonl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a71cc5da8225d8d44148c13f94b769279d43e7ed1b0c8088db8eeff57bfb2ddb
+size 18863250

checkpoint-100/math.test.v1.1.0shot.metrics.json

@@ -0,0 +1,5 @@
+{
+  "acc": 0.15,
+  "correct": 750,
+  "total": 5000
+}

checkpoint-100/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d6bb6fcce4e50c2a97bb2ba33ece1810e089aacf448e4ac57cd2357511787f96
+size 5012367854

checkpoint-100/special_tokens_map.json

@@ -0,0 +1,34 @@
+{
+  "additional_special_tokens": [
+    "<start_of_turn>",
+    "<end_of_turn>"
+  ],
+  "bos_token": {
+    "content": "<bos>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "<eos>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<pad>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

checkpoint-100/tokenizer.json

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:05e97791a5e007260de1db7e1692e53150e08cea481e2bf25435553380c147ee
+size 17477929

checkpoint-100/tokenizer.model

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:61a7b147390c64585d6c3543dd6fc636906c9af3865a5548f27f31aee1d4c8e2
+size 4241003

checkpoint-100/tokenizer_config.json

@@ -0,0 +1,70 @@
+{
+  "add_bos_token": true,
+  "add_eos_token": false,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<pad>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<eos>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "<bos>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "3": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "106": {
+      "content": "<start_of_turn>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "107": {
+      "content": "<end_of_turn>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [
+    "<start_of_turn>",
+    "<end_of_turn>"
+  ],
+  "bos_token": "<bos>",
+  "chat_template": "{% if messages[0]['role'] == 'system' %}{{ raise_exception('System role not supported') }}{% endif %}{% for message in messages %}{% if (message['role'] == 'user') != (loop.index0 % 2 == 0) %}{{ raise_exception('Conversation roles must alternate user/assistant/user/assistant/...') }}{% endif %}{% if (message['role'] == 'assistant') %}{% set role = 'model' %}{% else %}{% set role = message['role'] %}{% endif %}{{ '<start_of_turn>' + role + '\n' + message['content'] | trim + '<end_of_turn>\n' }}{% endfor %}{% if add_generation_prompt %}{{'<start_of_turn>model\n'}}{% endif %}",
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "<eos>",
+  "legacy": null,
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "<pad>",
+  "sp_model_kwargs": {},
+  "spaces_between_special_tokens": false,
+  "tokenizer_class": "GemmaTokenizer",
+  "unk_token": "<unk>",
+  "use_default_system_prompt": false
+}

checkpoint-100/training_config.yaml

@@ -0,0 +1,211 @@
+ds_cfg:
+  train_micro_batch_size_per_gpu: ${per_gpu_train_batch_size}
+  gradient_accumulation_steps: ${gradient_accumulation_steps}
+  scheduler:
+    type: WarmupDecayLR
+    params:
+      total_num_steps: 989
+      warmup_max_lr: ${learning_rate}
+      warmup_num_steps: 59
+      warmup_type: linear
+  optimizer:
+    type: AdamW
+    params:
+      lr: ${learning_rate}
+      betas:
+      - 0.9
+      - 0.95
+      eps: 1.0e-06
+      weight_decay: ${weight_decay}
+  bf16:
+    enabled: true
+  zero_optimization:
+    stage: 1
+    stage3_param_persistence_threshold: 100000.0
+    stage3_max_live_parameters: 100000000.0
+    stage3_prefetch_bucket_size: 100000000.0
+    memory_efficient_linear: false
+  steps_per_print: 25
+  gradient_clipping: 1.0
+  prescale_gradients: false
+sft_model_dir: experiments/gemma.2b.it.meta_math_distil.H100.w4.v1.0/checkpoint-2000/
+train_file: ${sft_model_dir}/meta_math_sub.25k.rap.train.0shot.n10.tem1.0.p0.7.v1.0_clean_fix.prm_cp3200_best_of_10.v1.0.(2,3).pos0.5.neg10.product.full_only_pos2pos.json
+dev_file: null
+test_file: null
+torch_dtype:
+  _target_: general_util.training_utils.return_torch_dtype
+  dtype: bfloat16
+tokenizer_init:
+  _target_: general_util.tokenization_utils.init_tokenizer
+  tokenizer_path: experiments/gemma.2b.it.meta_math_distil.H100.w4.v1.0/checkpoint-2000/
+  padding_side: left
+device_map:
+  _target_: models.utils.return_single_device_map
+model:
+  _target_: models.gemma.GemmaForCausalLMDPO.from_pretrained_with_ref_model
+  beta: 0.5
+  gradient_checkpointing: false
+  attn_implementation: flash_attention_2
+  torch_dtype: ${torch_dtype}
+  device_map: ${device_map}
+  ref_model:
+    _target_: models.gemma.GemmaForCausalLM.from_pretrained
+    pretrained_model_name_or_path: ${model_name_or_path}
+    torch_dtype: ${torch_dtype}
+    attn_implementation: flash_attention_2
+    device_map: ${device_map}
+read_tensor_dpo:
+  _target_: data.logic_combine.MultiMappingDataset
+  file_path: experiments/gemma.2b.it.meta_math_distil.H100.w4.v1.0/checkpoint-2000//meta_math_sub.25k.rap.train.0shot.n10.tem1.0.p0.7.v1.0_clean_fix.json
+  aligner:
+    _target_: data.input_aligner.concat_aligner
+    aligners:
+    - _target_: data.input_aligner.dpo_pair_aligner_cleaned
+      response_field: response
+      id_field: id
+      do_sample: false
+  template:
+    chosen: '{instruction}
+
+
+      ### Question: {query}
+
+
+      SubQuestion 1: {pos}<eos>'
+    reject: '{instruction}
+
+
+      ### Question: {query}
+
+
+      SubQuestion 1: {neg}<eos>'
+    prompt: '{instruction}
+
+
+      ### Question: {query}
+
+
+      SubQuestion 1:'
+  instruction: 'Given a question, please decompose it into sub-questions. For each
+    sub-question, please answer it in a complete sentence, ending with "The answer
+    is". When the original question is answerable, please start the sub-question with
+    "Now we can answer the question: ".'
+  kv_mapping:
+    chosen: chosen
+    reject: reject
+    id: index
+    prompt: prompt
+read_tensor_step_dpo:
+  _target_: data.logic_combine.MultiMappingDataset
+  aligner:
+    _target_: data.logic_combine.field_extract_aligner
+    input_index_field: id
+    extract_index_field: id
+    extract_fields:
+    - query
+    extra_file: experiments/gemma.2b.it.meta_math_distil.H100.w4.v1.0/checkpoint-2000//meta_math_sub.25k.rap.train.0shot.n10.tem1.0.p0.7.v1.0_clean_fix.json
+  template:
+    chosen: '{instruction}
+
+
+      ### Question: {query}
+
+
+      SubQuestion 1: {chosen}<eos>'
+    reject: '{instruction}
+
+
+      ### Question: {query}
+
+
+      SubQuestion 1: {reject}<eos>'
+    prompt: '{instruction}
+
+
+      ### Question: {query}
+
+
+      SubQuestion 1:'
+  instruction: 'Given a question, please decompose it into sub-questions. For each
+    sub-question, please answer it in a complete sentence, ending with "The answer
+    is". When the original question is answerable, please start the sub-question with
+    "Now we can answer the question: ".'
+  kv_mapping:
+    chosen: chosen
+    reject: reject
+    id: index
+    prompt: prompt
+read_tensor:
+  _target_: data.general.ReplayDataset
+  _recursive_: false
+  new_dataset_cfg: ${read_tensor_step_dpo}
+  old_dataset_cfg: ${read_tensor_dpo}
+  replay_ratio: 0.2
+dist_load_data_barrier: false
+extended_vocab: null
+collator:
+  _target_: data.dpo.DPOCollator
+  tokenizer: ${tokenizer_init}
+  max_seq_length: 1024
+num_workers: 8
+prefetch_factor: 2
+model_name_or_path: ${sft_model_dir}
+pretrain: null
+dp_size: 4
+tp_size: 1
+pp_size: 1
+exp_name: gemma.2b.it.meta_math_rap.step.dpo.H100.w4.v1.0.s${seed}
+exp_notes: null
+output_dir: experiments/${exp_name}
+do_train: true
+evaluate_during_training: false
+do_eval: false
+eval_sub_path: checkpoint-100
+per_gpu_train_batch_size: 2
+per_gpu_eval_batch_size: 4
+learning_rate: 1.0e-06
+gradient_accumulation_steps: 8
+weight_decay: 0.1
+adam_epsilon: 1.0e-06
+adam_betas: (0.9, 0.98)
+total_dataset_len: 63348
+max_grad_norm: 1.0
+num_train_epochs: 1
+max_steps: 0
+warmup_proportion: 0.06
+warmup_steps: 0
+optimizer: null
+use_nvlamb: null
+bit_training: null
+logging_steps: 5
+save_ds_state: false
+save_steps: 100
+save_best: false
+eval_steps: 400
+ddp_eval: true
+no_cuda: false
+seed: 43
+local_rank: 0
+fp16: true
+fp16_opt_level: O1
+fp16_bfloat16: true
+prediction_cfg:
+  metric: loss
+  measure: -1
+  best_checkpoint: null
+  best_result: null
+eval_forward_fn:
+  _target_: general_util.evaluator.DefaultForwardFn
+post_process:
+  _target_: post_processors.dpo.DPOEvalPostProcessor
+summary_helper:
+  _target_: general_util.tensorboard_helper.WandbWriter
+  batch_index_or_keys: null
+  outputs_index_or_keys:
+    train/chosen_reward: chosen_reward
+    train/rejected_reward: rejected_reward
+n_gpu: 1
+device: cuda:0
+train_batch_size: 2
+eval_batch_size: null
+world_size: 4

checkpoint-200/config.json

@@ -0,0 +1,28 @@
+{
+  "_name_or_path": "experiments/gemma.2b.it.meta_math_distil.H100.w4.v1.0/checkpoint-2000/",
+  "architectures": [
+    "GemmaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "bos_token_id": 2,
+  "eos_token_id": 1,
+  "head_dim": 256,
+  "hidden_act": "gelu",
+  "hidden_size": 2048,
+  "initializer_range": 0.02,
+  "intermediate_size": 16384,
+  "max_position_embeddings": 8192,
+  "model_type": "gemma",
+  "num_attention_heads": 8,
+  "num_hidden_layers": 18,
+  "num_key_value_heads": 1,
+  "pad_token_id": 0,
+  "rms_norm_eps": 1e-06,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.38.2",
+  "use_cache": true,
+  "vocab_size": 256000
+}

checkpoint-200/generation_config.json

@@ -0,0 +1,7 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 2,
+  "eos_token_id": 1,
+  "pad_token_id": 0,
+  "transformers_version": "4.38.2"
+}

checkpoint-200/gsm8k.test.v1.1.0shot.metrics.json

@@ -0,0 +1,5 @@
+{
+  "acc": 0.5170583775587566,
+  "correct": 682,
+  "total": 1319
+}

checkpoint-200/math.test.v1.1.0shot.jsonl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0a1d53ec71532a1b15e81828d9e4a466438ec23564bc6fb7e5cdffb82641eff0
+size 19888106

checkpoint-200/math.test.v1.1.0shot.metrics.json

@@ -0,0 +1,5 @@
+{
+  "acc": 0.1536,
+  "correct": 768,
+  "total": 5000
+}

checkpoint-200/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7eee440231421df61e23a106ea15d23ab5546186f73f635558ef0d7f179c857e
+size 5012367854

checkpoint-200/special_tokens_map.json

@@ -0,0 +1,34 @@
+{
+  "additional_special_tokens": [
+    "<start_of_turn>",
+    "<end_of_turn>"
+  ],
+  "bos_token": {
+    "content": "<bos>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "<eos>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<pad>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

checkpoint-200/tokenizer.json

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:05e97791a5e007260de1db7e1692e53150e08cea481e2bf25435553380c147ee
+size 17477929

checkpoint-200/tokenizer.model

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:61a7b147390c64585d6c3543dd6fc636906c9af3865a5548f27f31aee1d4c8e2
+size 4241003

checkpoint-200/tokenizer_config.json

@@ -0,0 +1,70 @@
+{
+  "add_bos_token": true,
+  "add_eos_token": false,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<pad>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<eos>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "<bos>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "3": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "106": {
+      "content": "<start_of_turn>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "107": {
+      "content": "<end_of_turn>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [
+    "<start_of_turn>",
+    "<end_of_turn>"
+  ],
+  "bos_token": "<bos>",
+  "chat_template": "{% if messages[0]['role'] == 'system' %}{{ raise_exception('System role not supported') }}{% endif %}{% for message in messages %}{% if (message['role'] == 'user') != (loop.index0 % 2 == 0) %}{{ raise_exception('Conversation roles must alternate user/assistant/user/assistant/...') }}{% endif %}{% if (message['role'] == 'assistant') %}{% set role = 'model' %}{% else %}{% set role = message['role'] %}{% endif %}{{ '<start_of_turn>' + role + '\n' + message['content'] | trim + '<end_of_turn>\n' }}{% endfor %}{% if add_generation_prompt %}{{'<start_of_turn>model\n'}}{% endif %}",
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "<eos>",
+  "legacy": null,
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "<pad>",
+  "sp_model_kwargs": {},
+  "spaces_between_special_tokens": false,
+  "tokenizer_class": "GemmaTokenizer",
+  "unk_token": "<unk>",
+  "use_default_system_prompt": false
+}

checkpoint-200/training_config.yaml

@@ -0,0 +1,211 @@
+ds_cfg:
+  train_micro_batch_size_per_gpu: ${per_gpu_train_batch_size}
+  gradient_accumulation_steps: ${gradient_accumulation_steps}
+  scheduler:
+    type: WarmupDecayLR
+    params:
+      total_num_steps: 989
+      warmup_max_lr: ${learning_rate}
+      warmup_num_steps: 59
+      warmup_type: linear
+  optimizer:
+    type: AdamW
+    params:
+      lr: ${learning_rate}
+      betas:
+      - 0.9
+      - 0.95
+      eps: 1.0e-06
+      weight_decay: ${weight_decay}
+  bf16:
+    enabled: true
+  zero_optimization:
+    stage: 1
+    stage3_param_persistence_threshold: 100000.0
+    stage3_max_live_parameters: 100000000.0
+    stage3_prefetch_bucket_size: 100000000.0
+    memory_efficient_linear: false
+  steps_per_print: 25
+  gradient_clipping: 1.0
+  prescale_gradients: false
+sft_model_dir: experiments/gemma.2b.it.meta_math_distil.H100.w4.v1.0/checkpoint-2000/
+train_file: ${sft_model_dir}/meta_math_sub.25k.rap.train.0shot.n10.tem1.0.p0.7.v1.0_clean_fix.prm_cp3200_best_of_10.v1.0.(2,3).pos0.5.neg10.product.full_only_pos2pos.json
+dev_file: null
+test_file: null
+torch_dtype:
+  _target_: general_util.training_utils.return_torch_dtype
+  dtype: bfloat16
+tokenizer_init:
+  _target_: general_util.tokenization_utils.init_tokenizer
+  tokenizer_path: experiments/gemma.2b.it.meta_math_distil.H100.w4.v1.0/checkpoint-2000/
+  padding_side: left
+device_map:
+  _target_: models.utils.return_single_device_map
+model:
+  _target_: models.gemma.GemmaForCausalLMDPO.from_pretrained_with_ref_model
+  beta: 0.5
+  gradient_checkpointing: false
+  attn_implementation: flash_attention_2
+  torch_dtype: ${torch_dtype}
+  device_map: ${device_map}
+  ref_model:
+    _target_: models.gemma.GemmaForCausalLM.from_pretrained
+    pretrained_model_name_or_path: ${model_name_or_path}
+    torch_dtype: ${torch_dtype}
+    attn_implementation: flash_attention_2
+    device_map: ${device_map}
+read_tensor_dpo:
+  _target_: data.logic_combine.MultiMappingDataset
+  file_path: experiments/gemma.2b.it.meta_math_distil.H100.w4.v1.0/checkpoint-2000//meta_math_sub.25k.rap.train.0shot.n10.tem1.0.p0.7.v1.0_clean_fix.json
+  aligner:
+    _target_: data.input_aligner.concat_aligner
+    aligners:
+    - _target_: data.input_aligner.dpo_pair_aligner_cleaned
+      response_field: response
+      id_field: id
+      do_sample: false
+  template:
+    chosen: '{instruction}
+
+
+      ### Question: {query}
+
+
+      SubQuestion 1: {pos}<eos>'
+    reject: '{instruction}
+
+
+      ### Question: {query}
+
+
+      SubQuestion 1: {neg}<eos>'
+    prompt: '{instruction}
+
+
+      ### Question: {query}
+
+
+      SubQuestion 1:'
+  instruction: 'Given a question, please decompose it into sub-questions. For each
+    sub-question, please answer it in a complete sentence, ending with "The answer
+    is". When the original question is answerable, please start the sub-question with
+    "Now we can answer the question: ".'
+  kv_mapping:
+    chosen: chosen
+    reject: reject
+    id: index
+    prompt: prompt
+read_tensor_step_dpo:
+  _target_: data.logic_combine.MultiMappingDataset
+  aligner:
+    _target_: data.logic_combine.field_extract_aligner
+    input_index_field: id
+    extract_index_field: id
+    extract_fields:
+    - query
+    extra_file: experiments/gemma.2b.it.meta_math_distil.H100.w4.v1.0/checkpoint-2000//meta_math_sub.25k.rap.train.0shot.n10.tem1.0.p0.7.v1.0_clean_fix.json
+  template:
+    chosen: '{instruction}
+
+
+      ### Question: {query}
+
+
+      SubQuestion 1: {chosen}<eos>'
+    reject: '{instruction}
+
+
+      ### Question: {query}
+
+
+      SubQuestion 1: {reject}<eos>'
+    prompt: '{instruction}
+
+
+      ### Question: {query}
+
+
+      SubQuestion 1:'
+  instruction: 'Given a question, please decompose it into sub-questions. For each
+    sub-question, please answer it in a complete sentence, ending with "The answer
+    is". When the original question is answerable, please start the sub-question with
+    "Now we can answer the question: ".'
+  kv_mapping:
+    chosen: chosen
+    reject: reject
+    id: index
+    prompt: prompt
+read_tensor:
+  _target_: data.general.ReplayDataset
+  _recursive_: false
+  new_dataset_cfg: ${read_tensor_step_dpo}
+  old_dataset_cfg: ${read_tensor_dpo}
+  replay_ratio: 0.2
+dist_load_data_barrier: false
+extended_vocab: null
+collator:
+  _target_: data.dpo.DPOCollator
+  tokenizer: ${tokenizer_init}
+  max_seq_length: 1024
+num_workers: 8
+prefetch_factor: 2
+model_name_or_path: ${sft_model_dir}
+pretrain: null
+dp_size: 4
+tp_size: 1
+pp_size: 1
+exp_name: gemma.2b.it.meta_math_rap.step.dpo.H100.w4.v1.0.s${seed}
+exp_notes: null
+output_dir: experiments/${exp_name}
+do_train: true
+evaluate_during_training: false
+do_eval: false
+eval_sub_path: checkpoint-100
+per_gpu_train_batch_size: 2
+per_gpu_eval_batch_size: 4
+learning_rate: 1.0e-06
+gradient_accumulation_steps: 8
+weight_decay: 0.1
+adam_epsilon: 1.0e-06
+adam_betas: (0.9, 0.98)
+total_dataset_len: 63348
+max_grad_norm: 1.0
+num_train_epochs: 1
+max_steps: 0
+warmup_proportion: 0.06
+warmup_steps: 0
+optimizer: null
+use_nvlamb: null
+bit_training: null
+logging_steps: 5
+save_ds_state: false
+save_steps: 100
+save_best: false
+eval_steps: 400
+ddp_eval: true
+no_cuda: false
+seed: 43
+local_rank: 0
+fp16: true
+fp16_opt_level: O1
+fp16_bfloat16: true
+prediction_cfg:
+  metric: loss
+  measure: -1
+  best_checkpoint: null
+  best_result: null
+eval_forward_fn:
+  _target_: general_util.evaluator.DefaultForwardFn
+post_process:
+  _target_: post_processors.dpo.DPOEvalPostProcessor
+summary_helper:
+  _target_: general_util.tensorboard_helper.WandbWriter
+  batch_index_or_keys: null
+  outputs_index_or_keys:
+    train/chosen_reward: chosen_reward
+    train/rejected_reward: rejected_reward
+n_gpu: 1
+device: cuda:0
+train_batch_size: 2
+eval_batch_size: null
+world_size: 4

checkpoint-300/config.json

@@ -0,0 +1,28 @@
+{
+  "_name_or_path": "experiments/gemma.2b.it.meta_math_distil.H100.w4.v1.0/checkpoint-2000/",
+  "architectures": [
+    "GemmaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "bos_token_id": 2,
+  "eos_token_id": 1,
+  "head_dim": 256,
+  "hidden_act": "gelu",
+  "hidden_size": 2048,
+  "initializer_range": 0.02,
+  "intermediate_size": 16384,
+  "max_position_embeddings": 8192,
+  "model_type": "gemma",
+  "num_attention_heads": 8,
+  "num_hidden_layers": 18,
+  "num_key_value_heads": 1,
+  "pad_token_id": 0,
+  "rms_norm_eps": 1e-06,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.38.2",
+  "use_cache": true,
+  "vocab_size": 256000
+}

checkpoint-300/generation_config.json

@@ -0,0 +1,7 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 2,
+  "eos_token_id": 1,
+  "pad_token_id": 0,
+  "transformers_version": "4.38.2"
+}

checkpoint-300/gsm8k.test.v1.1.0shot.metrics.json

@@ -0,0 +1,5 @@
+{
+  "acc": 0.5261561789234268,
+  "correct": 694,
+  "total": 1319
+}

checkpoint-300/math.test.v1.1.0shot.json

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:77b0a19307fb3fb8482ba835251ee99f900ee543012b9321a1b7fffba12b472f
+size 10854292

checkpoint-300/math.test.v1.1.0shot.jsonl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8872901998a94c0184e4a12807565c81e05248fc24dfce24270eaba8c3f44cb4
+size 21438580

checkpoint-300/math.test.v1.1.0shot.metrics.json

@@ -0,0 +1,5 @@
+{
+  "acc": 0.1594,
+  "correct": 797,
+  "total": 5000
+}

checkpoint-300/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ae6ad37e169b44e1ed90724ef05f80000af38987f3aabd09255504b4bd889ef8
+size 5012367854

checkpoint-300/special_tokens_map.json

@@ -0,0 +1,34 @@
+{
+  "additional_special_tokens": [
+    "<start_of_turn>",
+    "<end_of_turn>"
+  ],
+  "bos_token": {
+    "content": "<bos>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "<eos>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<pad>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

checkpoint-300/tokenizer.json

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:05e97791a5e007260de1db7e1692e53150e08cea481e2bf25435553380c147ee
+size 17477929

checkpoint-300/tokenizer.model

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:61a7b147390c64585d6c3543dd6fc636906c9af3865a5548f27f31aee1d4c8e2
+size 4241003

checkpoint-300/tokenizer_config.json

@@ -0,0 +1,70 @@
+{
+  "add_bos_token": true,
+  "add_eos_token": false,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<pad>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<eos>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "<bos>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "3": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "106": {
+      "content": "<start_of_turn>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "107": {
+      "content": "<end_of_turn>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [
+    "<start_of_turn>",
+    "<end_of_turn>"
+  ],
+  "bos_token": "<bos>",
+  "chat_template": "{% if messages[0]['role'] == 'system' %}{{ raise_exception('System role not supported') }}{% endif %}{% for message in messages %}{% if (message['role'] == 'user') != (loop.index0 % 2 == 0) %}{{ raise_exception('Conversation roles must alternate user/assistant/user/assistant/...') }}{% endif %}{% if (message['role'] == 'assistant') %}{% set role = 'model' %}{% else %}{% set role = message['role'] %}{% endif %}{{ '<start_of_turn>' + role + '\n' + message['content'] | trim + '<end_of_turn>\n' }}{% endfor %}{% if add_generation_prompt %}{{'<start_of_turn>model\n'}}{% endif %}",
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "<eos>",
+  "legacy": null,
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "<pad>",
+  "sp_model_kwargs": {},
+  "spaces_between_special_tokens": false,
+  "tokenizer_class": "GemmaTokenizer",
+  "unk_token": "<unk>",
+  "use_default_system_prompt": false
+}

checkpoint-300/training_config.yaml

@@ -0,0 +1,211 @@
+ds_cfg:
+  train_micro_batch_size_per_gpu: ${per_gpu_train_batch_size}
+  gradient_accumulation_steps: ${gradient_accumulation_steps}
+  scheduler:
+    type: WarmupDecayLR
+    params:
+      total_num_steps: 989
+      warmup_max_lr: ${learning_rate}
+      warmup_num_steps: 59
+      warmup_type: linear
+  optimizer:
+    type: AdamW
+    params:
+      lr: ${learning_rate}
+      betas:
+      - 0.9
+      - 0.95
+      eps: 1.0e-06
+      weight_decay: ${weight_decay}
+  bf16:
+    enabled: true
+  zero_optimization:
+    stage: 1
+    stage3_param_persistence_threshold: 100000.0
+    stage3_max_live_parameters: 100000000.0
+    stage3_prefetch_bucket_size: 100000000.0
+    memory_efficient_linear: false
+  steps_per_print: 25
+  gradient_clipping: 1.0
+  prescale_gradients: false
+sft_model_dir: experiments/gemma.2b.it.meta_math_distil.H100.w4.v1.0/checkpoint-2000/
+train_file: ${sft_model_dir}/meta_math_sub.25k.rap.train.0shot.n10.tem1.0.p0.7.v1.0_clean_fix.prm_cp3200_best_of_10.v1.0.(2,3).pos0.5.neg10.product.full_only_pos2pos.json
+dev_file: null
+test_file: null
+torch_dtype:
+  _target_: general_util.training_utils.return_torch_dtype
+  dtype: bfloat16
+tokenizer_init:
+  _target_: general_util.tokenization_utils.init_tokenizer
+  tokenizer_path: experiments/gemma.2b.it.meta_math_distil.H100.w4.v1.0/checkpoint-2000/
+  padding_side: left
+device_map:
+  _target_: models.utils.return_single_device_map
+model:
+  _target_: models.gemma.GemmaForCausalLMDPO.from_pretrained_with_ref_model
+  beta: 0.5
+  gradient_checkpointing: false
+  attn_implementation: flash_attention_2
+  torch_dtype: ${torch_dtype}
+  device_map: ${device_map}
+  ref_model:
+    _target_: models.gemma.GemmaForCausalLM.from_pretrained
+    pretrained_model_name_or_path: ${model_name_or_path}
+    torch_dtype: ${torch_dtype}
+    attn_implementation: flash_attention_2
+    device_map: ${device_map}
+read_tensor_dpo:
+  _target_: data.logic_combine.MultiMappingDataset
+  file_path: experiments/gemma.2b.it.meta_math_distil.H100.w4.v1.0/checkpoint-2000//meta_math_sub.25k.rap.train.0shot.n10.tem1.0.p0.7.v1.0_clean_fix.json
+  aligner:
+    _target_: data.input_aligner.concat_aligner
+    aligners:
+    - _target_: data.input_aligner.dpo_pair_aligner_cleaned
+      response_field: response
+      id_field: id
+      do_sample: false
+  template:
+    chosen: '{instruction}
+
+
+      ### Question: {query}
+
+
+      SubQuestion 1: {pos}<eos>'
+    reject: '{instruction}
+
+
+      ### Question: {query}
+
+
+      SubQuestion 1: {neg}<eos>'
+    prompt: '{instruction}
+
+
+      ### Question: {query}
+
+
+      SubQuestion 1:'
+  instruction: 'Given a question, please decompose it into sub-questions. For each
+    sub-question, please answer it in a complete sentence, ending with "The answer
+    is". When the original question is answerable, please start the sub-question with
+    "Now we can answer the question: ".'
+  kv_mapping:
+    chosen: chosen
+    reject: reject
+    id: index
+    prompt: prompt
+read_tensor_step_dpo:
+  _target_: data.logic_combine.MultiMappingDataset
+  aligner:
+    _target_: data.logic_combine.field_extract_aligner
+    input_index_field: id
+    extract_index_field: id
+    extract_fields:
+    - query
+    extra_file: experiments/gemma.2b.it.meta_math_distil.H100.w4.v1.0/checkpoint-2000//meta_math_sub.25k.rap.train.0shot.n10.tem1.0.p0.7.v1.0_clean_fix.json
+  template:
+    chosen: '{instruction}
+
+
+      ### Question: {query}
+
+
+      SubQuestion 1: {chosen}<eos>'
+    reject: '{instruction}
+
+
+      ### Question: {query}
+
+
+      SubQuestion 1: {reject}<eos>'
+    prompt: '{instruction}
+
+
+      ### Question: {query}
+
+
+      SubQuestion 1:'
+  instruction: 'Given a question, please decompose it into sub-questions. For each
+    sub-question, please answer it in a complete sentence, ending with "The answer
+    is". When the original question is answerable, please start the sub-question with
+    "Now we can answer the question: ".'
+  kv_mapping:
+    chosen: chosen
+    reject: reject
+    id: index
+    prompt: prompt
+read_tensor:
+  _target_: data.general.ReplayDataset
+  _recursive_: false
+  new_dataset_cfg: ${read_tensor_step_dpo}
+  old_dataset_cfg: ${read_tensor_dpo}
+  replay_ratio: 0.2
+dist_load_data_barrier: false
+extended_vocab: null
+collator:
+  _target_: data.dpo.DPOCollator
+  tokenizer: ${tokenizer_init}
+  max_seq_length: 1024
+num_workers: 8
+prefetch_factor: 2
+model_name_or_path: ${sft_model_dir}
+pretrain: null
+dp_size: 4
+tp_size: 1
+pp_size: 1
+exp_name: gemma.2b.it.meta_math_rap.step.dpo.H100.w4.v1.0.s${seed}
+exp_notes: null
+output_dir: experiments/${exp_name}
+do_train: true
+evaluate_during_training: false
+do_eval: false
+eval_sub_path: checkpoint-100
+per_gpu_train_batch_size: 2
+per_gpu_eval_batch_size: 4
+learning_rate: 1.0e-06
+gradient_accumulation_steps: 8
+weight_decay: 0.1
+adam_epsilon: 1.0e-06
+adam_betas: (0.9, 0.98)
+total_dataset_len: 63348
+max_grad_norm: 1.0
+num_train_epochs: 1
+max_steps: 0
+warmup_proportion: 0.06
+warmup_steps: 0
+optimizer: null
+use_nvlamb: null
+bit_training: null
+logging_steps: 5
+save_ds_state: false
+save_steps: 100
+save_best: false
+eval_steps: 400
+ddp_eval: true
+no_cuda: false
+seed: 43
+local_rank: 0
+fp16: true
+fp16_opt_level: O1
+fp16_bfloat16: true
+prediction_cfg:
+  metric: loss
+  measure: -1
+  best_checkpoint: null
+  best_result: null
+eval_forward_fn:
+  _target_: general_util.evaluator.DefaultForwardFn
+post_process:
+  _target_: post_processors.dpo.DPOEvalPostProcessor
+summary_helper:
+  _target_: general_util.tensorboard_helper.WandbWriter
+  batch_index_or_keys: null
+  outputs_index_or_keys:
+    train/chosen_reward: chosen_reward
+    train/rejected_reward: rejected_reward
+n_gpu: 1
+device: cuda:0
+train_batch_size: 2
+eval_batch_size: null
+world_size: 4

checkpoint-400/config.json

@@ -0,0 +1,28 @@
+{
+  "_name_or_path": "experiments/gemma.2b.it.meta_math_distil.H100.w4.v1.0/checkpoint-2000/",
+  "architectures": [
+    "GemmaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "bos_token_id": 2,
+  "eos_token_id": 1,
+  "head_dim": 256,
+  "hidden_act": "gelu",
+  "hidden_size": 2048,
+  "initializer_range": 0.02,
+  "intermediate_size": 16384,
+  "max_position_embeddings": 8192,
+  "model_type": "gemma",
+  "num_attention_heads": 8,
+  "num_hidden_layers": 18,
+  "num_key_value_heads": 1,
+  "pad_token_id": 0,
+  "rms_norm_eps": 1e-06,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.38.2",
+  "use_cache": true,
+  "vocab_size": 256000
+}

checkpoint-400/generation_config.json

@@ -0,0 +1,7 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 2,
+  "eos_token_id": 1,
+  "pad_token_id": 0,
+  "transformers_version": "4.38.2"
+}

checkpoint-400/gsm8k.test.v1.1.0shot.metrics.json

@@ -0,0 +1,5 @@
+{
+  "acc": 0.5200909780136467,
+  "correct": 686,
+  "total": 1319
+}

checkpoint-400/math.test.v1.1.0shot.json

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6f15ce99022a48d9acb6f9ce9588fec9468144e9b72ac88522fdb388a42a3918
+size 11489913

checkpoint-400/math.test.v1.1.0shot.jsonl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6cc017f82fd3d5386cdd82e01350e8a6a1ee345a54440327515aa616d2c53200
+size 22709822