Model save

README.md

@@ -0,0 +1,60 @@
+---
+license: llama3
+base_model: Jackie999/llama3-sudo-5epochs-tofu_full_sft
+tags:
+- trl
+- dpo
+- generated_from_trainer
+model-index:
+- name: llama3-sudo-dpo-5epochs-forget10mix400-1sft-2fullpara-2e-4
+  results: []
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+# llama3-sudo-dpo-5epochs-forget10mix400-1sft-2fullpara-2e-4
+
+This model is a fine-tuned version of [Jackie999/llama3-sudo-5epochs-tofu_full_sft](https://huggingface.co/Jackie999/llama3-sudo-5epochs-tofu_full_sft) on an unknown dataset.
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 0.0002
+- train_batch_size: 4
+- eval_batch_size: 8
+- seed: 42
+- distributed_type: multi-GPU
+- num_devices: 4
+- gradient_accumulation_steps: 4
+- total_train_batch_size: 64
+- total_eval_batch_size: 32
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: cosine
+- lr_scheduler_warmup_ratio: 0.1
+- num_epochs: 5
+
+### Training results
+
+
+
+### Framework versions
+
+- Transformers 4.44.0
+- Pytorch 2.1.2
+- Datasets 3.0.0
+- Tokenizers 0.19.1

all_results.json

@@ -0,0 +1,9 @@
+{
+    "epoch": 5.0,
+    "total_flos": 0.0,
+    "train_loss": 0.7484294853210449,
+    "train_runtime": 12665.918,
+    "train_samples": 1600,
+    "train_samples_per_second": 0.632,
+    "train_steps_per_second": 0.01
+}

generation_config.json

@@ -0,0 +1,12 @@
+{
+  "bos_token_id": 128000,
+  "do_sample": true,
+  "eos_token_id": [
+    128001,
+    128009
+  ],
+  "max_length": 4096,
+  "temperature": 0.6,
+  "top_p": 0.9,
+  "transformers_version": "4.44.0"
+}

train_results.json

@@ -0,0 +1,9 @@
+{
+    "epoch": 5.0,
+    "total_flos": 0.0,
+    "train_loss": 0.7484294853210449,
+    "train_runtime": 12665.918,
+    "train_samples": 1600,
+    "train_samples_per_second": 0.632,
+    "train_steps_per_second": 0.01
+}

trainer_state.json

@@ -0,0 +1,237 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 5.0,
+  "eval_steps": 1000,
+  "global_step": 125,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.04,
+      "grad_norm": 5.49035456239289,
+      "learning_rate": 1.5384615384615387e-05,
+      "logits/chosen": -1.73323655128479,
+      "logits/rejected": -1.963712453842163,
+      "logps/chosen": -64.71795654296875,
+      "logps/rejected": -92.56527709960938,
+      "loss": 0.6931,
+      "rewards/accuracies": 0.0,
+      "rewards/chosen": 0.0,
+      "rewards/margins": 0.0,
+      "rewards/rejected": 0.0,
+      "step": 1
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 28.445585026780677,
+      "learning_rate": 0.00015384615384615385,
+      "logits/chosen": -1.9578092098236084,
+      "logits/rejected": -2.1086513996124268,
+      "logps/chosen": -241.50640869140625,
+      "logps/rejected": -268.8847351074219,
+      "loss": 1.1389,
+      "rewards/accuracies": 0.4722222089767456,
+      "rewards/chosen": -1.7811455726623535,
+      "rewards/margins": 0.029786672443151474,
+      "rewards/rejected": -1.8109323978424072,
+      "step": 10
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 5.507863124304794,
+      "learning_rate": 0.00019807852804032305,
+      "logits/chosen": -2.51621675491333,
+      "logits/rejected": -2.6432676315307617,
+      "logps/chosen": -570.351806640625,
+      "logps/rejected": -496.91827392578125,
+      "loss": 1.8146,
+      "rewards/accuracies": 0.35624998807907104,
+      "rewards/chosen": -5.092972755432129,
+      "rewards/margins": -1.0333130359649658,
+      "rewards/rejected": -4.059659957885742,
+      "step": 20
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.9111283957421439,
+      "learning_rate": 0.00018884456359788724,
+      "logits/chosen": -3.369929790496826,
+      "logits/rejected": -3.283642292022705,
+      "logps/chosen": -507.5963439941406,
+      "logps/rejected": -572.895263671875,
+      "loss": 0.7716,
+      "rewards/accuracies": 0.6187499761581421,
+      "rewards/chosen": -4.45947265625,
+      "rewards/margins": 0.3659425377845764,
+      "rewards/rejected": -4.825415134429932,
+      "step": 30
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 1.795612553020093,
+      "learning_rate": 0.0001726660322034027,
+      "logits/chosen": -4.420574188232422,
+      "logits/rejected": -4.248013496398926,
+      "logps/chosen": -548.0731811523438,
+      "logps/rejected": -683.3683471679688,
+      "loss": 0.639,
+      "rewards/accuracies": 0.7124999761581421,
+      "rewards/chosen": -4.906264305114746,
+      "rewards/margins": 0.9860424995422363,
+      "rewards/rejected": -5.892307758331299,
+      "step": 40
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 0.8538474858298817,
+      "learning_rate": 0.00015080753452465296,
+      "logits/chosen": -5.002673149108887,
+      "logits/rejected": -4.903468132019043,
+      "logps/chosen": -598.4771728515625,
+      "logps/rejected": -708.3795166015625,
+      "loss": 0.6112,
+      "rewards/accuracies": 0.706250011920929,
+      "rewards/chosen": -5.362724781036377,
+      "rewards/margins": 0.840530514717102,
+      "rewards/rejected": -6.203255653381348,
+      "step": 50
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 0.6914654231679778,
+      "learning_rate": 0.0001249776478167227,
+      "logits/chosen": -5.280072212219238,
+      "logits/rejected": -5.135374069213867,
+      "logps/chosen": -556.451904296875,
+      "logps/rejected": -724.9661254882812,
+      "loss": 0.494,
+      "rewards/accuracies": 0.731249988079071,
+      "rewards/chosen": -4.981103420257568,
+      "rewards/margins": 1.3216989040374756,
+      "rewards/rejected": -6.302803039550781,
+      "step": 60
+    },
+    {
+      "epoch": 2.8,
+      "grad_norm": 0.8698084153477929,
+      "learning_rate": 9.719537437241312e-05,
+      "logits/chosen": -5.114866733551025,
+      "logits/rejected": -5.01845121383667,
+      "logps/chosen": -614.8002319335938,
+      "logps/rejected": -744.7828369140625,
+      "loss": 0.5243,
+      "rewards/accuracies": 0.668749988079071,
+      "rewards/chosen": -5.515006065368652,
+      "rewards/margins": 1.0367931127548218,
+      "rewards/rejected": -6.5517988204956055,
+      "step": 70
+    },
+    {
+      "epoch": 3.2,
+      "grad_norm": 0.7549954414499753,
+      "learning_rate": 6.963232548903853e-05,
+      "logits/chosen": -4.472952365875244,
+      "logits/rejected": -4.432765007019043,
+      "logps/chosen": -761.5465087890625,
+      "logps/rejected": -860.75537109375,
+      "loss": 1.6723,
+      "rewards/accuracies": 0.675000011920929,
+      "rewards/chosen": -7.0114946365356445,
+      "rewards/margins": 0.6902900338172913,
+      "rewards/rejected": -7.701784610748291,
+      "step": 80
+    },
+    {
+      "epoch": 3.6,
+      "grad_norm": 1.1425715631983737,
+      "learning_rate": 4.444297669803981e-05,
+      "logits/chosen": -5.620628356933594,
+      "logits/rejected": -5.535118103027344,
+      "logps/chosen": -655.6566772460938,
+      "logps/rejected": -807.3014526367188,
+      "loss": 0.4738,
+      "rewards/accuracies": 0.737500011920929,
+      "rewards/chosen": -5.936182498931885,
+      "rewards/margins": 1.246588945388794,
+      "rewards/rejected": -7.1827712059021,
+      "step": 90
+    },
+    {
+      "epoch": 4.0,
+      "grad_norm": 1.9528369738814884,
+      "learning_rate": 2.3596262417839255e-05,
+      "logits/chosen": -5.550174236297607,
+      "logits/rejected": -5.511833667755127,
+      "logps/chosen": -628.8092041015625,
+      "logps/rejected": -796.935302734375,
+      "loss": 0.449,
+      "rewards/accuracies": 0.737500011920929,
+      "rewards/chosen": -5.668156623840332,
+      "rewards/margins": 1.3851256370544434,
+      "rewards/rejected": -7.053282260894775,
+      "step": 100
+    },
+    {
+      "epoch": 4.4,
+      "grad_norm": 2.890782481250775,
+      "learning_rate": 8.72167349386811e-06,
+      "logits/chosen": -5.907111167907715,
+      "logits/rejected": -5.945315361022949,
+      "logps/chosen": -695.7470092773438,
+      "logps/rejected": -886.4703369140625,
+      "loss": 0.3633,
+      "rewards/accuracies": 0.856249988079071,
+      "rewards/chosen": -6.3334550857543945,
+      "rewards/margins": 1.6471582651138306,
+      "rewards/rejected": -7.980613708496094,
+      "step": 110
+    },
+    {
+      "epoch": 4.8,
+      "grad_norm": 0.7286712015593821,
+      "learning_rate": 9.818874663554357e-07,
+      "logits/chosen": -5.903909683227539,
+      "logits/rejected": -5.952976226806641,
+      "logps/chosen": -664.2296752929688,
+      "logps/rejected": -862.1025390625,
+      "loss": 0.3324,
+      "rewards/accuracies": 0.90625,
+      "rewards/chosen": -6.032934665679932,
+      "rewards/margins": 1.685362458229065,
+      "rewards/rejected": -7.718297004699707,
+      "step": 120
+    },
+    {
+      "epoch": 5.0,
+      "step": 125,
+      "total_flos": 0.0,
+      "train_loss": 0.7484294853210449,
+      "train_runtime": 12665.918,
+      "train_samples_per_second": 0.632,
+      "train_steps_per_second": 0.01
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 125,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 5,
+  "save_steps": 1000,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 0.0,
+  "train_batch_size": 4,
+  "trial_name": null,
+  "trial_params": null
+}