checkpoint-200 adapter (common-step selection)

checkpoint-200/README.md

@@ -0,0 +1,210 @@
+---
+base_model: allenai/Olmo-3-1025-7B
+library_name: peft
+pipeline_tag: text-generation
+tags:
+- base_model:adapter:allenai/Olmo-3-1025-7B
+- lora
+- sft
+- transformers
+- trl
+- unsloth
+---
+
+# Model Card for Model ID
+
+<!-- Provide a quick summary of what the model is/does. -->
+
+
+
+## Model Details
+
+### Model Description
+
+<!-- Provide a longer summary of what this model is. -->
+
+
+
+- **Developed by:** [More Information Needed]
+- **Funded by [optional]:** [More Information Needed]
+- **Shared by [optional]:** [More Information Needed]
+- **Model type:** [More Information Needed]
+- **Language(s) (NLP):** [More Information Needed]
+- **License:** [More Information Needed]
+- **Finetuned from model [optional]:** [More Information Needed]
+
+### Model Sources [optional]
+
+<!-- Provide the basic links for the model. -->
+
+- **Repository:** [More Information Needed]
+- **Paper [optional]:** [More Information Needed]
+- **Demo [optional]:** [More Information Needed]
+
+## Uses
+
+<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
+
+### Direct Use
+
+<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
+
+[More Information Needed]
+
+### Downstream Use [optional]
+
+<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
+
+[More Information Needed]
+
+### Out-of-Scope Use
+
+<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
+
+[More Information Needed]
+
+## Bias, Risks, and Limitations
+
+<!-- This section is meant to convey both technical and sociotechnical limitations. -->
+
+[More Information Needed]
+
+### Recommendations
+
+<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
+
+Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
+
+## How to Get Started with the Model
+
+Use the code below to get started with the model.
+
+[More Information Needed]
+
+## Training Details
+
+### Training Data
+
+<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
+
+[More Information Needed]
+
+### Training Procedure
+
+<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
+
+#### Preprocessing [optional]
+
+[More Information Needed]
+
+
+#### Training Hyperparameters
+
+- **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
+
+#### Speeds, Sizes, Times [optional]
+
+<!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
+
+[More Information Needed]
+
+## Evaluation
+
+<!-- This section describes the evaluation protocols and provides the results. -->
+
+### Testing Data, Factors & Metrics
+
+#### Testing Data
+
+<!-- This should link to a Dataset Card if possible. -->
+
+[More Information Needed]
+
+#### Factors
+
+<!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
+
+[More Information Needed]
+
+#### Metrics
+
+<!-- These are the evaluation metrics being used, ideally with a description of why. -->
+
+[More Information Needed]
+
+### Results
+
+[More Information Needed]
+
+#### Summary
+
+
+
+## Model Examination [optional]
+
+<!-- Relevant interpretability work for the model goes here -->
+
+[More Information Needed]
+
+## Environmental Impact
+
+<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
+
+Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
+
+- **Hardware Type:** [More Information Needed]
+- **Hours used:** [More Information Needed]
+- **Cloud Provider:** [More Information Needed]
+- **Compute Region:** [More Information Needed]
+- **Carbon Emitted:** [More Information Needed]
+
+## Technical Specifications [optional]
+
+### Model Architecture and Objective
+
+[More Information Needed]
+
+### Compute Infrastructure
+
+[More Information Needed]
+
+#### Hardware
+
+[More Information Needed]
+
+#### Software
+
+[More Information Needed]
+
+## Citation [optional]
+
+<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
+
+**BibTeX:**
+
+[More Information Needed]
+
+**APA:**
+
+[More Information Needed]
+
+## Glossary [optional]
+
+<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
+
+[More Information Needed]
+
+## More Information [optional]
+
+[More Information Needed]
+
+## Model Card Authors [optional]
+
+[More Information Needed]
+
+## Model Card Contact
+
+[More Information Needed]
+### Framework versions
+
+- PEFT 0.20.0

checkpoint-200/adapter_config.json

@@ -0,0 +1,49 @@
+{
+  "alora_invocation_tokens": null,
+  "alpha_pattern": {},
+  "arrow_config": null,
+  "auto_mapping": {
+    "base_model_class": "Olmo3ForCausalLM",
+    "parent_library": "transformers.models.olmo3.modeling_olmo3",
+    "unsloth_fixed": true
+  },
+  "base_model_name_or_path": "allenai/Olmo-3-1025-7B",
+  "bias": "none",
+  "corda_config": null,
+  "ensure_weight_tying": false,
+  "eva_config": null,
+  "exclude_modules": null,
+  "fan_in_fan_out": false,
+  "inference_mode": true,
+  "init_lora_weights": true,
+  "layer_replication": null,
+  "layers_pattern": null,
+  "layers_to_transform": null,
+  "loftq_config": {},
+  "lora_alpha": 128,
+  "lora_bias": false,
+  "lora_dropout": 0.05,
+  "lora_ga_config": null,
+  "megatron_config": null,
+  "megatron_core": "megatron.core",
+  "modules_to_save": [
+    "embed_tokens",
+    "lm_head"
+  ],
+  "monteclora_config": null,
+  "peft_type": "LORA",
+  "peft_version": "0.20.0",
+  "qalora_group_size": 16,
+  "r": 64,
+  "rank_pattern": {},
+  "revision": null,
+  "target_modules": "(?:.*?(?:language|text).*?(?:self_attn|attention|attn|mixer|mlp|feed_forward|ffn|dense|mixer).*?(?:q_proj|k_proj|v_proj|o_proj|gate_proj|up_proj|down_proj))|(?:\\bmodel\\.layers\\.[\\d]{1,}\\.(?:self_attn|attention|attn|mixer|mlp|feed_forward|ffn|dense|mixer)\\.(?:(?:q_proj|k_proj|v_proj|o_proj|gate_proj|up_proj|down_proj)))",
+  "target_parameters": null,
+  "task_type": "CAUSAL_LM",
+  "trainable_token_indices": null,
+  "use_bdlora": null,
+  "use_dora": false,
+  "use_qalora": false,
+  "use_rslora": false,
+  "velora_config": null
+}

checkpoint-200/adapter_model.safetensors

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

checkpoint-200/chat_template.jinja

@@ -0,0 +1,16 @@
+{%- set has_system = messages|selectattr('role', 'equalto', 'system')|list|length > 0 -%}{%- if not has_system -%}{{- '<|im_start|>system
+You are a helpful function-calling AI assistant. ' -}}{%- if tools is none -%}{{- 'You do not currently have access to any functions. <functions></functions><|im_end|>
+' -}}{%- else -%}{{- 'You are provided with function signatures within <functions></functions> XML tags. You may call one or more functions to assist with the user query. Output any function calls within <function_calls></function_calls> XML tags. Do not make assumptions about what values to plug into functions.' -}}{{- '<functions>' -}}{{- tools | tojson -}}{{- '</functions><|im_end|>
+' -}}{%- endif -%}{%- endif -%}{%- for message in messages -%}{%- if message['role'] == 'system' -%}{{- '<|im_start|>system
+' + message['content'] -}}{%- if tools is not none -%}{{- '<functions>' -}}{{- tools | tojson -}}{{- '</functions>' -}}{%- elif message.get('functions', none) is not none -%}{{- ' <functions>' + message['functions'] + '</functions>' -}}{%- endif -%}{{- '<|im_end|>
+' -}}{%- elif message['role'] == 'user' -%}{{- '<|im_start|>user
+' + message['content'] + '<|im_end|>
+' -}}{%- elif message['role'] == 'assistant' -%}{{- '<|im_start|>assistant
+' -}}{%- if message.get('content', none) is not none -%}{{- message['content'] -}}{%- endif -%}{%- if message.get('function_calls', none) is not none -%}{{- '<function_calls>' + message['function_calls'] + '</function_calls>' -}}{% elif message.get('tool_calls', none) is not none %}{{- '<function_calls>' -}}{%- for tool_call in message['tool_calls'] %}{%- if tool_call is mapping and tool_call.get('function', none) is not none %}{%- set args = tool_call['function']['arguments'] -%}{%- set ns = namespace(arguments_list=[]) -%}{%- for key, value in args.items() -%}{%- set ns.arguments_list = ns.arguments_list + [key ~ '=' ~ (value | tojson)] -%}{%- endfor -%}{%- set arguments = ns.arguments_list | join(', ') -%}{{- tool_call['function']['name'] + '(' + arguments + ')' -}}{%- if not loop.last -%}{{ '
+' }}{%- endif -%}{% else %}{{- tool_call -}}{%- endif %}{%- endfor %}{{- '</function_calls>' -}}{%- endif -%}{%- if not loop.last -%}{{- '<|im_end|>' + '
+' -}}{%- else -%}{{- eos_token -}}{%- endif -%}{%- elif message['role'] == 'environment' -%}{{- '<|im_start|>environment
+' + message['content'] + '<|im_end|>
+' -}}{%- elif message['role'] == 'tool' -%}{{- '<|im_start|>environment
+' + message['content'] + '<|im_end|>
+' -}}{%- endif -%}{%- if loop.last and add_generation_prompt -%}{{- '<|im_start|>assistant
+' -}}{%- endif -%}{%- endfor -%}

checkpoint-200/special_tokens_map.json

@@ -0,0 +1,23 @@
+{
+  "eos_token": {
+    "content": "<|endoftext|>",
+    "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": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

checkpoint-200/tokenizer_config.json

@@ -0,0 +1,190 @@
+{
+  "add_prefix_space": false,
+  "added_tokens_decoder": {
+    "100256": {
+      "content": "<|extra_id_0|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "100257": {
+      "content": "<|endoftext|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "100258": {
+      "content": "<|fim_prefix|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "100259": {
+      "content": "<|fim_middle|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "100260": {
+      "content": "<|fim_suffix|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "100261": {
+      "content": "|||PHONE_NUMBER|||",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "100262": {
+      "content": "|||EMAIL_ADDRESS|||",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "100263": {
+      "content": "|||IP_ADDRESS|||",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "100264": {
+      "content": "<|im_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "100265": {
+      "content": "<|im_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "100266": {
+      "content": "<|extra_id_1|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "100267": {
+      "content": "<|extra_id_2|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "100268": {
+      "content": "<|extra_id_3|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "100269": {
+      "content": "<|extra_id_4|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "100270": {
+      "content": "<|extra_id_5|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "100271": {
+      "content": "<|extra_id_6|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "100272": {
+      "content": "<|extra_id_7|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "100273": {
+      "content": "<|extra_id_8|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "100274": {
+      "content": "<|extra_id_9|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "100275": {
+      "content": "<|extra_id_10|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "100276": {
+      "content": "<|endofprompt|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "100277": {
+      "content": "<|pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "bos_token": null,
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "<|endoftext|>",
+  "extra_special_tokens": {},
+  "model_max_length": 65536,
+  "pad_token": "<|pad|>",
+  "padding_side": "right",
+  "tokenizer_class": "GPT2Tokenizer",
+  "unk_token": "<|endoftext|>"
+}

checkpoint-200/trainer_state.json

@@ -0,0 +1,355 @@
+{
+  "best_global_step": 200,
+  "best_metric": 0.14798370003700256,
+  "best_model_checkpoint": "/workspace/anchor/olmo-pretrained/train/checkpoint-200",
+  "epoch": 0.2321532211259431,
+  "eval_steps": 50,
+  "global_step": 200,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.005803830528148578,
+      "grad_norm": 1.8992259502410889,
+      "learning_rate": 1.8181818181818182e-05,
+      "loss": 0.9941,
+      "step": 5
+    },
+    {
+      "epoch": 0.011607661056297156,
+      "grad_norm": 1.5108100175857544,
+      "learning_rate": 4.0909090909090915e-05,
+      "loss": 0.8727,
+      "step": 10
+    },
+    {
+      "epoch": 0.017411491584445733,
+      "grad_norm": 1.313894271850586,
+      "learning_rate": 6.363636363636364e-05,
+      "loss": 0.7164,
+      "step": 15
+    },
+    {
+      "epoch": 0.023215322112594312,
+      "grad_norm": 1.5354955196380615,
+      "learning_rate": 8.636363636363637e-05,
+      "loss": 0.5324,
+      "step": 20
+    },
+    {
+      "epoch": 0.02901915264074289,
+      "grad_norm": 1.4390828609466553,
+      "learning_rate": 0.00010909090909090909,
+      "loss": 0.4492,
+      "step": 25
+    },
+    {
+      "epoch": 0.034822983168891465,
+      "grad_norm": 1.8266173601150513,
+      "learning_rate": 0.0001318181818181818,
+      "loss": 0.4172,
+      "step": 30
+    },
+    {
+      "epoch": 0.04062681369704005,
+      "grad_norm": 1.5203430652618408,
+      "learning_rate": 0.00015454545454545454,
+      "loss": 0.3966,
+      "step": 35
+    },
+    {
+      "epoch": 0.046430644225188625,
+      "grad_norm": 1.6018481254577637,
+      "learning_rate": 0.00017727272727272728,
+      "loss": 0.3584,
+      "step": 40
+    },
+    {
+      "epoch": 0.0522344747533372,
+      "grad_norm": 1.3058902025222778,
+      "learning_rate": 0.0002,
+      "loss": 0.3571,
+      "step": 45
+    },
+    {
+      "epoch": 0.05803830528148578,
+      "grad_norm": 1.388102650642395,
+      "learning_rate": 0.00019998156301911284,
+      "loss": 0.2975,
+      "step": 50
+    },
+    {
+      "epoch": 0.05803830528148578,
+      "eval_loss": 0.2423536628484726,
+      "eval_runtime": 152.5048,
+      "eval_samples_per_second": 2.321,
+      "eval_steps_per_second": 1.161,
+      "step": 50
+    },
+    {
+      "epoch": 0.06384213580963435,
+      "grad_norm": 1.3211257457733154,
+      "learning_rate": 0.0001999262588748966,
+      "loss": 0.3098,
+      "step": 55
+    },
+    {
+      "epoch": 0.06964596633778293,
+      "grad_norm": 1.0737800598144531,
+      "learning_rate": 0.00019983410796018032,
+      "loss": 0.27,
+      "step": 60
+    },
+    {
+      "epoch": 0.07544979686593152,
+      "grad_norm": 1.062996745109558,
+      "learning_rate": 0.00019970514425465705,
+      "loss": 0.2597,
+      "step": 65
+    },
+    {
+      "epoch": 0.0812536273940801,
+      "grad_norm": 0.8997095227241516,
+      "learning_rate": 0.00019953941531235424,
+      "loss": 0.2213,
+      "step": 70
+    },
+    {
+      "epoch": 0.08705745792222867,
+      "grad_norm": 0.6930145025253296,
+      "learning_rate": 0.00019933698224409876,
+      "loss": 0.2616,
+      "step": 75
+    },
+    {
+      "epoch": 0.09286128845037725,
+      "grad_norm": 0.5634636282920837,
+      "learning_rate": 0.0001990979196949828,
+      "loss": 0.2218,
+      "step": 80
+    },
+    {
+      "epoch": 0.09866511897852583,
+      "grad_norm": 0.545459508895874,
+      "learning_rate": 0.00019882231581683937,
+      "loss": 0.238,
+      "step": 85
+    },
+    {
+      "epoch": 0.1044689495066744,
+      "grad_norm": 0.4912775456905365,
+      "learning_rate": 0.0001985102722357371,
+      "loss": 0.2145,
+      "step": 90
+    },
+    {
+      "epoch": 0.11027278003482298,
+      "grad_norm": 0.4108012914657593,
+      "learning_rate": 0.0001981619040145068,
+      "loss": 0.2336,
+      "step": 95
+    },
+    {
+      "epoch": 0.11607661056297155,
+      "grad_norm": 0.521858274936676,
+      "learning_rate": 0.00019777733961031326,
+      "loss": 0.1963,
+      "step": 100
+    },
+    {
+      "epoch": 0.11607661056297155,
+      "eval_loss": 0.1702147275209427,
+      "eval_runtime": 57.6184,
+      "eval_samples_per_second": 6.144,
+      "eval_steps_per_second": 3.072,
+      "step": 100
+    },
+    {
+      "epoch": 0.12188044109112015,
+      "grad_norm": 0.5323580503463745,
+      "learning_rate": 0.00019735672082728782,
+      "loss": 0.2433,
+      "step": 105
+    },
+    {
+      "epoch": 0.1276842716192687,
+      "grad_norm": 0.5639608502388,
+      "learning_rate": 0.0001969002027642398,
+      "loss": 0.2063,
+      "step": 110
+    },
+    {
+      "epoch": 0.1334881021474173,
+      "grad_norm": 0.583233654499054,
+      "learning_rate": 0.00019640795375746518,
+      "loss": 0.2208,
+      "step": 115
+    },
+    {
+      "epoch": 0.13929193267556586,
+      "grad_norm": 0.5018272399902344,
+      "learning_rate": 0.00019588015531867464,
+      "loss": 0.204,
+      "step": 120
+    },
+    {
+      "epoch": 0.14509576320371445,
+      "grad_norm": 0.48238405585289,
+      "learning_rate": 0.00019531700206806274,
+      "loss": 0.2155,
+      "step": 125
+    },
+    {
+      "epoch": 0.15089959373186304,
+      "grad_norm": 0.4539037048816681,
+      "learning_rate": 0.00019471870166254377,
+      "loss": 0.208,
+      "step": 130
+    },
+    {
+      "epoch": 0.1567034242600116,
+      "grad_norm": 0.5225255489349365,
+      "learning_rate": 0.00019408547471918061,
+      "loss": 0.2292,
+      "step": 135
+    },
+    {
+      "epoch": 0.1625072547881602,
+      "grad_norm": 0.4645092785358429,
+      "learning_rate": 0.00019341755473383432,
+      "loss": 0.2074,
+      "step": 140
+    },
+    {
+      "epoch": 0.16831108531630876,
+      "grad_norm": 0.47805657982826233,
+      "learning_rate": 0.00019271518799506492,
+      "loss": 0.2301,
+      "step": 145
+    },
+    {
+      "epoch": 0.17411491584445735,
+      "grad_norm": 0.43033567070961,
+      "learning_rate": 0.00019197863349331522,
+      "loss": 0.1916,
+      "step": 150
+    },
+    {
+      "epoch": 0.17411491584445735,
+      "eval_loss": 0.1569332629442215,
+      "eval_runtime": 57.9274,
+      "eval_samples_per_second": 6.111,
+      "eval_steps_per_second": 3.056,
+      "step": 150
+    },
+    {
+      "epoch": 0.1799187463726059,
+      "grad_norm": 0.5715239644050598,
+      "learning_rate": 0.00019120816282541063,
+      "loss": 0.2054,
+      "step": 155
+    },
+    {
+      "epoch": 0.1857225769007545,
+      "grad_norm": 0.5529009699821472,
+      "learning_rate": 0.00019040406009441073,
+      "loss": 0.2156,
+      "step": 160
+    },
+    {
+      "epoch": 0.1915264074289031,
+      "grad_norm": 0.5217124819755554,
+      "learning_rate": 0.00018956662180484913,
+      "loss": 0.227,
+      "step": 165
+    },
+    {
+      "epoch": 0.19733023795705165,
+      "grad_norm": 0.4713853895664215,
+      "learning_rate": 0.00018869615675340068,
+      "loss": 0.2191,
+      "step": 170
+    },
+    {
+      "epoch": 0.20313406848520024,
+      "grad_norm": 0.44989559054374695,
+      "learning_rate": 0.00018779298591501564,
+      "loss": 0.2052,
+      "step": 175
+    },
+    {
+      "epoch": 0.2089378990133488,
+      "grad_norm": 0.5454024076461792,
+      "learning_rate": 0.00018685744232456374,
+      "loss": 0.2009,
+      "step": 180
+    },
+    {
+      "epoch": 0.2147417295414974,
+      "grad_norm": 0.5243939161300659,
+      "learning_rate": 0.0001858898709540309,
+      "loss": 0.175,
+      "step": 185
+    },
+    {
+      "epoch": 0.22054556006964596,
+      "grad_norm": 0.4877641499042511,
+      "learning_rate": 0.0001848906285853145,
+      "loss": 0.1861,
+      "step": 190
+    },
+    {
+      "epoch": 0.22634939059779455,
+      "grad_norm": 0.45391249656677246,
+      "learning_rate": 0.0001838600836786635,
+      "loss": 0.1775,
+      "step": 195
+    },
+    {
+      "epoch": 0.2321532211259431,
+      "grad_norm": 0.49695494771003723,
+      "learning_rate": 0.0001827986162368129,
+      "loss": 0.1791,
+      "step": 200
+    },
+    {
+      "epoch": 0.2321532211259431,
+      "eval_loss": 0.14798370003700256,
+      "eval_runtime": 57.9291,
+      "eval_samples_per_second": 6.111,
+      "eval_steps_per_second": 3.055,
+      "step": 200
+    }
+  ],
+  "logging_steps": 5,
+  "max_steps": 862,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 1,
+  "save_steps": 100,
+  "stateful_callbacks": {
+    "EarlyStoppingCallback": {
+      "args": {
+        "early_stopping_patience": 2,
+        "early_stopping_threshold": 0.001
+      },
+      "attributes": {
+        "early_stopping_patience_counter": 0
+      }
+    },
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": false
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 5.601436507676344e+17,
+  "train_batch_size": 2,
+  "trial_name": null,
+  "trial_params": null
+}