Text Generation
Transformers
Safetensors
qwen2
llama-factory
full
Generated from Trainer
conversational
text-generation-inference
Instructions to use PatrickG1014/alfworld-1.5b-group_semantic-refinement with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use PatrickG1014/alfworld-1.5b-group_semantic-refinement with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="PatrickG1014/alfworld-1.5b-group_semantic-refinement") messages = [ {"role": "user", "content": "Who are you?"}, ] pipe(messages)# Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("PatrickG1014/alfworld-1.5b-group_semantic-refinement") model = AutoModelForCausalLM.from_pretrained("PatrickG1014/alfworld-1.5b-group_semantic-refinement") messages = [ {"role": "user", "content": "Who are you?"}, ] inputs = tokenizer.apply_chat_template( messages, add_generation_prompt=True, tokenize=True, return_dict=True, return_tensors="pt", ).to(model.device) outputs = model.generate(**inputs, max_new_tokens=40) print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:])) - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use PatrickG1014/alfworld-1.5b-group_semantic-refinement with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "PatrickG1014/alfworld-1.5b-group_semantic-refinement" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "PatrickG1014/alfworld-1.5b-group_semantic-refinement", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker
docker model run hf.co/PatrickG1014/alfworld-1.5b-group_semantic-refinement
- SGLang
How to use PatrickG1014/alfworld-1.5b-group_semantic-refinement with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "PatrickG1014/alfworld-1.5b-group_semantic-refinement" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "PatrickG1014/alfworld-1.5b-group_semantic-refinement", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "PatrickG1014/alfworld-1.5b-group_semantic-refinement" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "PatrickG1014/alfworld-1.5b-group_semantic-refinement", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }' - Docker Model Runner
How to use PatrickG1014/alfworld-1.5b-group_semantic-refinement with Docker Model Runner:
docker model run hf.co/PatrickG1014/alfworld-1.5b-group_semantic-refinement
Upload folder using huggingface_hub
Browse files- .gitattributes +1 -0
- README.md +61 -0
- added_tokens.json +24 -0
- all_results.json +8 -0
- chat_template.jinja +54 -0
- config.json +28 -0
- generation_config.json +14 -0
- merges.txt +0 -0
- model.safetensors +3 -0
- special_tokens_map.json +31 -0
- tokenizer.json +3 -0
- tokenizer_config.json +212 -0
- train_results.json +8 -0
- trainer_log.jsonl +76 -0
- trainer_state.json +568 -0
- training_args.bin +3 -0
- training_loss.png +0 -0
- vocab.json +0 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
|
@@ -0,0 +1,61 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: transformers
|
| 3 |
+
license: other
|
| 4 |
+
base_model: Qwen/Qwen2.5-1.5B-Instruct
|
| 5 |
+
tags:
|
| 6 |
+
- llama-factory
|
| 7 |
+
- full
|
| 8 |
+
- generated_from_trainer
|
| 9 |
+
model-index:
|
| 10 |
+
- name: refinement
|
| 11 |
+
results: []
|
| 12 |
+
---
|
| 13 |
+
|
| 14 |
+
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
| 15 |
+
should probably proofread and complete it, then remove this comment. -->
|
| 16 |
+
|
| 17 |
+
# refinement
|
| 18 |
+
|
| 19 |
+
This model is a fine-tuned version of [/root/paddlejob/workspace/env_run/output/models/alfworld-1.5b-group_semantic](https://huggingface.co//root/paddlejob/workspace/env_run/output/models/alfworld-1.5b-group_semantic) on the refinement dataset.
|
| 20 |
+
|
| 21 |
+
## Model description
|
| 22 |
+
|
| 23 |
+
More information needed
|
| 24 |
+
|
| 25 |
+
## Intended uses & limitations
|
| 26 |
+
|
| 27 |
+
More information needed
|
| 28 |
+
|
| 29 |
+
## Training and evaluation data
|
| 30 |
+
|
| 31 |
+
More information needed
|
| 32 |
+
|
| 33 |
+
## Training procedure
|
| 34 |
+
|
| 35 |
+
### Training hyperparameters
|
| 36 |
+
|
| 37 |
+
The following hyperparameters were used during training:
|
| 38 |
+
- learning_rate: 1e-05
|
| 39 |
+
- train_batch_size: 1
|
| 40 |
+
- eval_batch_size: 8
|
| 41 |
+
- seed: 42
|
| 42 |
+
- distributed_type: multi-GPU
|
| 43 |
+
- num_devices: 2
|
| 44 |
+
- gradient_accumulation_steps: 2
|
| 45 |
+
- total_train_batch_size: 4
|
| 46 |
+
- total_eval_batch_size: 16
|
| 47 |
+
- optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
|
| 48 |
+
- lr_scheduler_type: cosine
|
| 49 |
+
- lr_scheduler_warmup_ratio: 0.1
|
| 50 |
+
- num_epochs: 3.0
|
| 51 |
+
|
| 52 |
+
### Training results
|
| 53 |
+
|
| 54 |
+
|
| 55 |
+
|
| 56 |
+
### Framework versions
|
| 57 |
+
|
| 58 |
+
- Transformers 4.52.4
|
| 59 |
+
- Pytorch 2.7.1+cu126
|
| 60 |
+
- Datasets 3.6.0
|
| 61 |
+
- Tokenizers 0.21.1
|
added_tokens.json
ADDED
|
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"</tool_call>": 151658,
|
| 3 |
+
"<tool_call>": 151657,
|
| 4 |
+
"<|box_end|>": 151649,
|
| 5 |
+
"<|box_start|>": 151648,
|
| 6 |
+
"<|endoftext|>": 151643,
|
| 7 |
+
"<|file_sep|>": 151664,
|
| 8 |
+
"<|fim_middle|>": 151660,
|
| 9 |
+
"<|fim_pad|>": 151662,
|
| 10 |
+
"<|fim_prefix|>": 151659,
|
| 11 |
+
"<|fim_suffix|>": 151661,
|
| 12 |
+
"<|im_end|>": 151645,
|
| 13 |
+
"<|im_start|>": 151644,
|
| 14 |
+
"<|image_pad|>": 151655,
|
| 15 |
+
"<|object_ref_end|>": 151647,
|
| 16 |
+
"<|object_ref_start|>": 151646,
|
| 17 |
+
"<|quad_end|>": 151651,
|
| 18 |
+
"<|quad_start|>": 151650,
|
| 19 |
+
"<|repo_name|>": 151663,
|
| 20 |
+
"<|video_pad|>": 151656,
|
| 21 |
+
"<|vision_end|>": 151653,
|
| 22 |
+
"<|vision_pad|>": 151654,
|
| 23 |
+
"<|vision_start|>": 151652
|
| 24 |
+
}
|
all_results.json
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"epoch": 3.0,
|
| 3 |
+
"total_flos": 5335736082432.0,
|
| 4 |
+
"train_loss": 0.30388372325897217,
|
| 5 |
+
"train_runtime": 989.6325,
|
| 6 |
+
"train_samples_per_second": 3.031,
|
| 7 |
+
"train_steps_per_second": 0.758
|
| 8 |
+
}
|
chat_template.jinja
ADDED
|
@@ -0,0 +1,54 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{%- if tools %}
|
| 2 |
+
{{- '<|im_start|>system\n' }}
|
| 3 |
+
{%- if messages[0]['role'] == 'system' %}
|
| 4 |
+
{{- messages[0]['content'] }}
|
| 5 |
+
{%- else %}
|
| 6 |
+
{{- 'You are Qwen, created by Alibaba Cloud. You are a helpful assistant.' }}
|
| 7 |
+
{%- endif %}
|
| 8 |
+
{{- "\n\n# Tools\n\nYou may call one or more functions to assist with the user query.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>" }}
|
| 9 |
+
{%- for tool in tools %}
|
| 10 |
+
{{- "\n" }}
|
| 11 |
+
{{- tool | tojson }}
|
| 12 |
+
{%- endfor %}
|
| 13 |
+
{{- "\n</tools>\n\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\n<tool_call>\n{\"name\": <function-name>, \"arguments\": <args-json-object>}\n</tool_call><|im_end|>\n" }}
|
| 14 |
+
{%- else %}
|
| 15 |
+
{%- if messages[0]['role'] == 'system' %}
|
| 16 |
+
{{- '<|im_start|>system\n' + messages[0]['content'] + '<|im_end|>\n' }}
|
| 17 |
+
{%- else %}
|
| 18 |
+
{{- '<|im_start|>system\nYou are Qwen, created by Alibaba Cloud. You are a helpful assistant.<|im_end|>\n' }}
|
| 19 |
+
{%- endif %}
|
| 20 |
+
{%- endif %}
|
| 21 |
+
{%- for message in messages %}
|
| 22 |
+
{%- if (message.role == "user") or (message.role == "system" and not loop.first) or (message.role == "assistant" and not message.tool_calls) %}
|
| 23 |
+
{{- '<|im_start|>' + message.role + '\n' + message.content + '<|im_end|>' + '\n' }}
|
| 24 |
+
{%- elif message.role == "assistant" %}
|
| 25 |
+
{{- '<|im_start|>' + message.role }}
|
| 26 |
+
{%- if message.content %}
|
| 27 |
+
{{- '\n' + message.content }}
|
| 28 |
+
{%- endif %}
|
| 29 |
+
{%- for tool_call in message.tool_calls %}
|
| 30 |
+
{%- if tool_call.function is defined %}
|
| 31 |
+
{%- set tool_call = tool_call.function %}
|
| 32 |
+
{%- endif %}
|
| 33 |
+
{{- '\n<tool_call>\n{"name": "' }}
|
| 34 |
+
{{- tool_call.name }}
|
| 35 |
+
{{- '", "arguments": ' }}
|
| 36 |
+
{{- tool_call.arguments | tojson }}
|
| 37 |
+
{{- '}\n</tool_call>' }}
|
| 38 |
+
{%- endfor %}
|
| 39 |
+
{{- '<|im_end|>\n' }}
|
| 40 |
+
{%- elif message.role == "tool" %}
|
| 41 |
+
{%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != "tool") %}
|
| 42 |
+
{{- '<|im_start|>user' }}
|
| 43 |
+
{%- endif %}
|
| 44 |
+
{{- '\n<tool_response>\n' }}
|
| 45 |
+
{{- message.content }}
|
| 46 |
+
{{- '\n</tool_response>' }}
|
| 47 |
+
{%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
|
| 48 |
+
{{- '<|im_end|>\n' }}
|
| 49 |
+
{%- endif %}
|
| 50 |
+
{%- endif %}
|
| 51 |
+
{%- endfor %}
|
| 52 |
+
{%- if add_generation_prompt %}
|
| 53 |
+
{{- '<|im_start|>assistant\n' }}
|
| 54 |
+
{%- endif %}
|
config.json
ADDED
|
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"Qwen2ForCausalLM"
|
| 4 |
+
],
|
| 5 |
+
"attention_dropout": 0.0,
|
| 6 |
+
"bos_token_id": 151643,
|
| 7 |
+
"eos_token_id": 151645,
|
| 8 |
+
"hidden_act": "silu",
|
| 9 |
+
"hidden_size": 1536,
|
| 10 |
+
"initializer_range": 0.02,
|
| 11 |
+
"intermediate_size": 8960,
|
| 12 |
+
"max_position_embeddings": 32768,
|
| 13 |
+
"max_window_layers": 21,
|
| 14 |
+
"model_type": "qwen2",
|
| 15 |
+
"num_attention_heads": 12,
|
| 16 |
+
"num_hidden_layers": 28,
|
| 17 |
+
"num_key_value_heads": 2,
|
| 18 |
+
"rms_norm_eps": 1e-06,
|
| 19 |
+
"rope_scaling": null,
|
| 20 |
+
"rope_theta": 1000000.0,
|
| 21 |
+
"sliding_window": 32768,
|
| 22 |
+
"tie_word_embeddings": true,
|
| 23 |
+
"torch_dtype": "bfloat16",
|
| 24 |
+
"transformers_version": "4.52.4",
|
| 25 |
+
"use_cache": false,
|
| 26 |
+
"use_sliding_window": false,
|
| 27 |
+
"vocab_size": 151936
|
| 28 |
+
}
|
generation_config.json
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"bos_token_id": 151643,
|
| 3 |
+
"do_sample": true,
|
| 4 |
+
"eos_token_id": [
|
| 5 |
+
151645,
|
| 6 |
+
151643
|
| 7 |
+
],
|
| 8 |
+
"pad_token_id": 151643,
|
| 9 |
+
"repetition_penalty": 1.1,
|
| 10 |
+
"temperature": 0.7,
|
| 11 |
+
"top_k": 20,
|
| 12 |
+
"top_p": 0.8,
|
| 13 |
+
"transformers_version": "4.52.4"
|
| 14 |
+
}
|
merges.txt
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ee85d219df61922c0cf2da8bfc1ad3ddee284ed8656c04fcd774a09dfd332521
|
| 3 |
+
size 3087467144
|
special_tokens_map.json
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"additional_special_tokens": [
|
| 3 |
+
"<|im_start|>",
|
| 4 |
+
"<|im_end|>",
|
| 5 |
+
"<|object_ref_start|>",
|
| 6 |
+
"<|object_ref_end|>",
|
| 7 |
+
"<|box_start|>",
|
| 8 |
+
"<|box_end|>",
|
| 9 |
+
"<|quad_start|>",
|
| 10 |
+
"<|quad_end|>",
|
| 11 |
+
"<|vision_start|>",
|
| 12 |
+
"<|vision_end|>",
|
| 13 |
+
"<|vision_pad|>",
|
| 14 |
+
"<|image_pad|>",
|
| 15 |
+
"<|video_pad|>"
|
| 16 |
+
],
|
| 17 |
+
"eos_token": {
|
| 18 |
+
"content": "<|im_end|>",
|
| 19 |
+
"lstrip": false,
|
| 20 |
+
"normalized": false,
|
| 21 |
+
"rstrip": false,
|
| 22 |
+
"single_word": false
|
| 23 |
+
},
|
| 24 |
+
"pad_token": {
|
| 25 |
+
"content": "<|endoftext|>",
|
| 26 |
+
"lstrip": false,
|
| 27 |
+
"normalized": false,
|
| 28 |
+
"rstrip": false,
|
| 29 |
+
"single_word": false
|
| 30 |
+
}
|
| 31 |
+
}
|
tokenizer.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:33624f49f1034c4f5d92e3ec47ccdf80ccd02caf1eb1d769872ba7fb1e5be112
|
| 3 |
+
size 11421995
|
tokenizer_config.json
ADDED
|
@@ -0,0 +1,212 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"add_bos_token": false,
|
| 3 |
+
"add_prefix_space": false,
|
| 4 |
+
"added_tokens_decoder": {
|
| 5 |
+
"151643": {
|
| 6 |
+
"content": "<|endoftext|>",
|
| 7 |
+
"lstrip": false,
|
| 8 |
+
"normalized": false,
|
| 9 |
+
"rstrip": false,
|
| 10 |
+
"single_word": false,
|
| 11 |
+
"special": true
|
| 12 |
+
},
|
| 13 |
+
"151644": {
|
| 14 |
+
"content": "<|im_start|>",
|
| 15 |
+
"lstrip": false,
|
| 16 |
+
"normalized": false,
|
| 17 |
+
"rstrip": false,
|
| 18 |
+
"single_word": false,
|
| 19 |
+
"special": true
|
| 20 |
+
},
|
| 21 |
+
"151645": {
|
| 22 |
+
"content": "<|im_end|>",
|
| 23 |
+
"lstrip": false,
|
| 24 |
+
"normalized": false,
|
| 25 |
+
"rstrip": false,
|
| 26 |
+
"single_word": false,
|
| 27 |
+
"special": true
|
| 28 |
+
},
|
| 29 |
+
"151646": {
|
| 30 |
+
"content": "<|object_ref_start|>",
|
| 31 |
+
"lstrip": false,
|
| 32 |
+
"normalized": false,
|
| 33 |
+
"rstrip": false,
|
| 34 |
+
"single_word": false,
|
| 35 |
+
"special": true
|
| 36 |
+
},
|
| 37 |
+
"151647": {
|
| 38 |
+
"content": "<|object_ref_end|>",
|
| 39 |
+
"lstrip": false,
|
| 40 |
+
"normalized": false,
|
| 41 |
+
"rstrip": false,
|
| 42 |
+
"single_word": false,
|
| 43 |
+
"special": true
|
| 44 |
+
},
|
| 45 |
+
"151648": {
|
| 46 |
+
"content": "<|box_start|>",
|
| 47 |
+
"lstrip": false,
|
| 48 |
+
"normalized": false,
|
| 49 |
+
"rstrip": false,
|
| 50 |
+
"single_word": false,
|
| 51 |
+
"special": true
|
| 52 |
+
},
|
| 53 |
+
"151649": {
|
| 54 |
+
"content": "<|box_end|>",
|
| 55 |
+
"lstrip": false,
|
| 56 |
+
"normalized": false,
|
| 57 |
+
"rstrip": false,
|
| 58 |
+
"single_word": false,
|
| 59 |
+
"special": true
|
| 60 |
+
},
|
| 61 |
+
"151650": {
|
| 62 |
+
"content": "<|quad_start|>",
|
| 63 |
+
"lstrip": false,
|
| 64 |
+
"normalized": false,
|
| 65 |
+
"rstrip": false,
|
| 66 |
+
"single_word": false,
|
| 67 |
+
"special": true
|
| 68 |
+
},
|
| 69 |
+
"151651": {
|
| 70 |
+
"content": "<|quad_end|>",
|
| 71 |
+
"lstrip": false,
|
| 72 |
+
"normalized": false,
|
| 73 |
+
"rstrip": false,
|
| 74 |
+
"single_word": false,
|
| 75 |
+
"special": true
|
| 76 |
+
},
|
| 77 |
+
"151652": {
|
| 78 |
+
"content": "<|vision_start|>",
|
| 79 |
+
"lstrip": false,
|
| 80 |
+
"normalized": false,
|
| 81 |
+
"rstrip": false,
|
| 82 |
+
"single_word": false,
|
| 83 |
+
"special": true
|
| 84 |
+
},
|
| 85 |
+
"151653": {
|
| 86 |
+
"content": "<|vision_end|>",
|
| 87 |
+
"lstrip": false,
|
| 88 |
+
"normalized": false,
|
| 89 |
+
"rstrip": false,
|
| 90 |
+
"single_word": false,
|
| 91 |
+
"special": true
|
| 92 |
+
},
|
| 93 |
+
"151654": {
|
| 94 |
+
"content": "<|vision_pad|>",
|
| 95 |
+
"lstrip": false,
|
| 96 |
+
"normalized": false,
|
| 97 |
+
"rstrip": false,
|
| 98 |
+
"single_word": false,
|
| 99 |
+
"special": true
|
| 100 |
+
},
|
| 101 |
+
"151655": {
|
| 102 |
+
"content": "<|image_pad|>",
|
| 103 |
+
"lstrip": false,
|
| 104 |
+
"normalized": false,
|
| 105 |
+
"rstrip": false,
|
| 106 |
+
"single_word": false,
|
| 107 |
+
"special": true
|
| 108 |
+
},
|
| 109 |
+
"151656": {
|
| 110 |
+
"content": "<|video_pad|>",
|
| 111 |
+
"lstrip": false,
|
| 112 |
+
"normalized": false,
|
| 113 |
+
"rstrip": false,
|
| 114 |
+
"single_word": false,
|
| 115 |
+
"special": true
|
| 116 |
+
},
|
| 117 |
+
"151657": {
|
| 118 |
+
"content": "<tool_call>",
|
| 119 |
+
"lstrip": false,
|
| 120 |
+
"normalized": false,
|
| 121 |
+
"rstrip": false,
|
| 122 |
+
"single_word": false,
|
| 123 |
+
"special": false
|
| 124 |
+
},
|
| 125 |
+
"151658": {
|
| 126 |
+
"content": "</tool_call>",
|
| 127 |
+
"lstrip": false,
|
| 128 |
+
"normalized": false,
|
| 129 |
+
"rstrip": false,
|
| 130 |
+
"single_word": false,
|
| 131 |
+
"special": false
|
| 132 |
+
},
|
| 133 |
+
"151659": {
|
| 134 |
+
"content": "<|fim_prefix|>",
|
| 135 |
+
"lstrip": false,
|
| 136 |
+
"normalized": false,
|
| 137 |
+
"rstrip": false,
|
| 138 |
+
"single_word": false,
|
| 139 |
+
"special": false
|
| 140 |
+
},
|
| 141 |
+
"151660": {
|
| 142 |
+
"content": "<|fim_middle|>",
|
| 143 |
+
"lstrip": false,
|
| 144 |
+
"normalized": false,
|
| 145 |
+
"rstrip": false,
|
| 146 |
+
"single_word": false,
|
| 147 |
+
"special": false
|
| 148 |
+
},
|
| 149 |
+
"151661": {
|
| 150 |
+
"content": "<|fim_suffix|>",
|
| 151 |
+
"lstrip": false,
|
| 152 |
+
"normalized": false,
|
| 153 |
+
"rstrip": false,
|
| 154 |
+
"single_word": false,
|
| 155 |
+
"special": false
|
| 156 |
+
},
|
| 157 |
+
"151662": {
|
| 158 |
+
"content": "<|fim_pad|>",
|
| 159 |
+
"lstrip": false,
|
| 160 |
+
"normalized": false,
|
| 161 |
+
"rstrip": false,
|
| 162 |
+
"single_word": false,
|
| 163 |
+
"special": false
|
| 164 |
+
},
|
| 165 |
+
"151663": {
|
| 166 |
+
"content": "<|repo_name|>",
|
| 167 |
+
"lstrip": false,
|
| 168 |
+
"normalized": false,
|
| 169 |
+
"rstrip": false,
|
| 170 |
+
"single_word": false,
|
| 171 |
+
"special": false
|
| 172 |
+
},
|
| 173 |
+
"151664": {
|
| 174 |
+
"content": "<|file_sep|>",
|
| 175 |
+
"lstrip": false,
|
| 176 |
+
"normalized": false,
|
| 177 |
+
"rstrip": false,
|
| 178 |
+
"single_word": false,
|
| 179 |
+
"special": false
|
| 180 |
+
}
|
| 181 |
+
},
|
| 182 |
+
"additional_special_tokens": [
|
| 183 |
+
"<|im_start|>",
|
| 184 |
+
"<|im_end|>",
|
| 185 |
+
"<|object_ref_start|>",
|
| 186 |
+
"<|object_ref_end|>",
|
| 187 |
+
"<|box_start|>",
|
| 188 |
+
"<|box_end|>",
|
| 189 |
+
"<|quad_start|>",
|
| 190 |
+
"<|quad_end|>",
|
| 191 |
+
"<|vision_start|>",
|
| 192 |
+
"<|vision_end|>",
|
| 193 |
+
"<|vision_pad|>",
|
| 194 |
+
"<|image_pad|>",
|
| 195 |
+
"<|video_pad|>"
|
| 196 |
+
],
|
| 197 |
+
"bos_token": null,
|
| 198 |
+
"clean_up_tokenization_spaces": false,
|
| 199 |
+
"eos_token": "<|im_end|>",
|
| 200 |
+
"errors": "replace",
|
| 201 |
+
"extra_special_tokens": {},
|
| 202 |
+
"max_length": 8192,
|
| 203 |
+
"model_max_length": 131072,
|
| 204 |
+
"pad_token": "<|endoftext|>",
|
| 205 |
+
"padding_side": "right",
|
| 206 |
+
"split_special_tokens": false,
|
| 207 |
+
"stride": 0,
|
| 208 |
+
"tokenizer_class": "Qwen2Tokenizer",
|
| 209 |
+
"truncation_side": "right",
|
| 210 |
+
"truncation_strategy": "longest_first",
|
| 211 |
+
"unk_token": null
|
| 212 |
+
}
|
train_results.json
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"epoch": 3.0,
|
| 3 |
+
"total_flos": 5335736082432.0,
|
| 4 |
+
"train_loss": 0.30388372325897217,
|
| 5 |
+
"train_runtime": 989.6325,
|
| 6 |
+
"train_samples_per_second": 3.031,
|
| 7 |
+
"train_steps_per_second": 0.758
|
| 8 |
+
}
|
trainer_log.jsonl
ADDED
|
@@ -0,0 +1,76 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{"current_steps": 10, "total_steps": 750, "loss": 1.0456, "lr": 1.2000000000000002e-06, "epoch": 0.04, "percentage": 1.33, "elapsed_time": "0:00:15", "remaining_time": "0:19:02"}
|
| 2 |
+
{"current_steps": 20, "total_steps": 750, "loss": 0.901, "lr": 2.5333333333333338e-06, "epoch": 0.08, "percentage": 2.67, "elapsed_time": "0:00:27", "remaining_time": "0:16:46"}
|
| 3 |
+
{"current_steps": 30, "total_steps": 750, "loss": 0.743, "lr": 3.866666666666667e-06, "epoch": 0.12, "percentage": 4.0, "elapsed_time": "0:00:39", "remaining_time": "0:15:54"}
|
| 4 |
+
{"current_steps": 40, "total_steps": 750, "loss": 0.5085, "lr": 5.2e-06, "epoch": 0.16, "percentage": 5.33, "elapsed_time": "0:00:52", "remaining_time": "0:15:24"}
|
| 5 |
+
{"current_steps": 50, "total_steps": 750, "loss": 0.5229, "lr": 6.533333333333334e-06, "epoch": 0.2, "percentage": 6.67, "elapsed_time": "0:01:04", "remaining_time": "0:15:03"}
|
| 6 |
+
{"current_steps": 60, "total_steps": 750, "loss": 0.4071, "lr": 7.866666666666667e-06, "epoch": 0.24, "percentage": 8.0, "elapsed_time": "0:01:16", "remaining_time": "0:14:42"}
|
| 7 |
+
{"current_steps": 70, "total_steps": 750, "loss": 0.3846, "lr": 9.200000000000002e-06, "epoch": 0.28, "percentage": 9.33, "elapsed_time": "0:01:28", "remaining_time": "0:14:23"}
|
| 8 |
+
{"current_steps": 80, "total_steps": 750, "loss": 0.4199, "lr": 9.99913355769784e-06, "epoch": 0.32, "percentage": 10.67, "elapsed_time": "0:01:41", "remaining_time": "0:14:09"}
|
| 9 |
+
{"current_steps": 90, "total_steps": 750, "loss": 0.4515, "lr": 9.989389530100242e-06, "epoch": 0.36, "percentage": 12.0, "elapsed_time": "0:01:53", "remaining_time": "0:13:54"}
|
| 10 |
+
{"current_steps": 100, "total_steps": 750, "loss": 0.4596, "lr": 9.968839595802982e-06, "epoch": 0.4, "percentage": 13.33, "elapsed_time": "0:02:06", "remaining_time": "0:13:39"}
|
| 11 |
+
{"current_steps": 110, "total_steps": 750, "loss": 0.4392, "lr": 9.937528261387753e-06, "epoch": 0.44, "percentage": 14.67, "elapsed_time": "0:02:18", "remaining_time": "0:13:25"}
|
| 12 |
+
{"current_steps": 120, "total_steps": 750, "loss": 0.4601, "lr": 9.89552334023258e-06, "epoch": 0.48, "percentage": 16.0, "elapsed_time": "0:02:31", "remaining_time": "0:13:13"}
|
| 13 |
+
{"current_steps": 130, "total_steps": 750, "loss": 0.4504, "lr": 9.842915805643156e-06, "epoch": 0.52, "percentage": 17.33, "elapsed_time": "0:02:43", "remaining_time": "0:13:01"}
|
| 14 |
+
{"current_steps": 140, "total_steps": 750, "loss": 0.4383, "lr": 9.779819593824909e-06, "epoch": 0.56, "percentage": 18.67, "elapsed_time": "0:02:55", "remaining_time": "0:12:46"}
|
| 15 |
+
{"current_steps": 150, "total_steps": 750, "loss": 0.4168, "lr": 9.70637135712256e-06, "epoch": 0.6, "percentage": 20.0, "elapsed_time": "0:03:08", "remaining_time": "0:12:32"}
|
| 16 |
+
{"current_steps": 160, "total_steps": 750, "loss": 0.4674, "lr": 9.622730168061568e-06, "epoch": 0.64, "percentage": 21.33, "elapsed_time": "0:03:20", "remaining_time": "0:12:19"}
|
| 17 |
+
{"current_steps": 170, "total_steps": 750, "loss": 0.4282, "lr": 9.529077174832466e-06, "epoch": 0.68, "percentage": 22.67, "elapsed_time": "0:03:32", "remaining_time": "0:12:05"}
|
| 18 |
+
{"current_steps": 180, "total_steps": 750, "loss": 0.4303, "lr": 9.425615208964217e-06, "epoch": 0.72, "percentage": 24.0, "elapsed_time": "0:03:44", "remaining_time": "0:11:52"}
|
| 19 |
+
{"current_steps": 190, "total_steps": 750, "loss": 0.429, "lr": 9.312568346036288e-06, "epoch": 0.76, "percentage": 25.33, "elapsed_time": "0:03:57", "remaining_time": "0:11:39"}
|
| 20 |
+
{"current_steps": 200, "total_steps": 750, "loss": 0.486, "lr": 9.190181420380838e-06, "epoch": 0.8, "percentage": 26.67, "elapsed_time": "0:04:09", "remaining_time": "0:11:27"}
|
| 21 |
+
{"current_steps": 210, "total_steps": 750, "loss": 0.3998, "lr": 9.058719494826076e-06, "epoch": 0.84, "percentage": 28.0, "elapsed_time": "0:04:22", "remaining_time": "0:11:16"}
|
| 22 |
+
{"current_steps": 220, "total_steps": 750, "loss": 0.4351, "lr": 8.9184672866292e-06, "epoch": 0.88, "percentage": 29.33, "elapsed_time": "0:04:35", "remaining_time": "0:11:03"}
|
| 23 |
+
{"current_steps": 230, "total_steps": 750, "loss": 0.4512, "lr": 8.769728550842217e-06, "epoch": 0.92, "percentage": 30.67, "elapsed_time": "0:04:48", "remaining_time": "0:10:51"}
|
| 24 |
+
{"current_steps": 240, "total_steps": 750, "loss": 0.421, "lr": 8.61282542244614e-06, "epoch": 0.96, "percentage": 32.0, "elapsed_time": "0:05:00", "remaining_time": "0:10:38"}
|
| 25 |
+
{"current_steps": 250, "total_steps": 750, "loss": 0.3976, "lr": 8.44809771867835e-06, "epoch": 1.0, "percentage": 33.33, "elapsed_time": "0:05:13", "remaining_time": "0:10:26"}
|
| 26 |
+
{"current_steps": 260, "total_steps": 750, "loss": 0.2612, "lr": 8.275902203064125e-06, "epoch": 1.04, "percentage": 34.67, "elapsed_time": "0:05:26", "remaining_time": "0:10:15"}
|
| 27 |
+
{"current_steps": 270, "total_steps": 750, "loss": 0.3022, "lr": 8.096611812746302e-06, "epoch": 1.08, "percentage": 36.0, "elapsed_time": "0:05:39", "remaining_time": "0:10:02"}
|
| 28 |
+
{"current_steps": 280, "total_steps": 750, "loss": 0.2954, "lr": 7.910614850786448e-06, "epoch": 1.12, "percentage": 37.33, "elapsed_time": "0:05:52", "remaining_time": "0:09:50"}
|
| 29 |
+
{"current_steps": 290, "total_steps": 750, "loss": 0.331, "lr": 7.718314145186918e-06, "epoch": 1.16, "percentage": 38.67, "elapsed_time": "0:06:04", "remaining_time": "0:09:38"}
|
| 30 |
+
{"current_steps": 300, "total_steps": 750, "loss": 0.2971, "lr": 7.520126176455084e-06, "epoch": 1.2, "percentage": 40.0, "elapsed_time": "0:06:17", "remaining_time": "0:09:26"}
|
| 31 |
+
{"current_steps": 310, "total_steps": 750, "loss": 0.2885, "lr": 7.31648017559931e-06, "epoch": 1.24, "percentage": 41.33, "elapsed_time": "0:06:29", "remaining_time": "0:09:13"}
|
| 32 |
+
{"current_steps": 320, "total_steps": 750, "loss": 0.2848, "lr": 7.107817194510157e-06, "epoch": 1.28, "percentage": 42.67, "elapsed_time": "0:06:42", "remaining_time": "0:09:00"}
|
| 33 |
+
{"current_steps": 330, "total_steps": 750, "loss": 0.2989, "lr": 6.8945891507402075e-06, "epoch": 1.32, "percentage": 44.0, "elapsed_time": "0:06:54", "remaining_time": "0:08:48"}
|
| 34 |
+
{"current_steps": 340, "total_steps": 750, "loss": 0.2939, "lr": 6.677257848751276e-06, "epoch": 1.3599999999999999, "percentage": 45.33, "elapsed_time": "0:07:07", "remaining_time": "0:08:35"}
|
| 35 |
+
{"current_steps": 350, "total_steps": 750, "loss": 0.2898, "lr": 6.456293979748778e-06, "epoch": 1.4, "percentage": 46.67, "elapsed_time": "0:07:20", "remaining_time": "0:08:23"}
|
| 36 |
+
{"current_steps": 360, "total_steps": 750, "loss": 0.2682, "lr": 6.23217610226939e-06, "epoch": 1.44, "percentage": 48.0, "elapsed_time": "0:07:32", "remaining_time": "0:08:10"}
|
| 37 |
+
{"current_steps": 370, "total_steps": 750, "loss": 0.2777, "lr": 6.005389605729824e-06, "epoch": 1.48, "percentage": 49.33, "elapsed_time": "0:07:45", "remaining_time": "0:07:57"}
|
| 38 |
+
{"current_steps": 380, "total_steps": 750, "loss": 0.268, "lr": 5.776425659181438e-06, "epoch": 1.52, "percentage": 50.67, "elapsed_time": "0:07:57", "remaining_time": "0:07:45"}
|
| 39 |
+
{"current_steps": 390, "total_steps": 750, "loss": 0.2945, "lr": 5.54578014754744e-06, "epoch": 1.56, "percentage": 52.0, "elapsed_time": "0:08:10", "remaining_time": "0:07:33"}
|
| 40 |
+
{"current_steps": 400, "total_steps": 750, "loss": 0.3164, "lr": 5.3139525976465675e-06, "epoch": 1.6, "percentage": 53.33, "elapsed_time": "0:08:23", "remaining_time": "0:07:20"}
|
| 41 |
+
{"current_steps": 410, "total_steps": 750, "loss": 0.262, "lr": 5.081445096329229e-06, "epoch": 1.6400000000000001, "percentage": 54.67, "elapsed_time": "0:08:36", "remaining_time": "0:07:08"}
|
| 42 |
+
{"current_steps": 420, "total_steps": 750, "loss": 0.2652, "lr": 4.8487612030691975e-06, "epoch": 1.6800000000000002, "percentage": 56.0, "elapsed_time": "0:08:48", "remaining_time": "0:06:55"}
|
| 43 |
+
{"current_steps": 430, "total_steps": 750, "loss": 0.2418, "lr": 4.6164048593659076e-06, "epoch": 1.72, "percentage": 57.33, "elapsed_time": "0:09:01", "remaining_time": "0:06:42"}
|
| 44 |
+
{"current_steps": 440, "total_steps": 750, "loss": 0.2714, "lr": 4.384879297319398e-06, "epoch": 1.76, "percentage": 58.67, "elapsed_time": "0:09:13", "remaining_time": "0:06:30"}
|
| 45 |
+
{"current_steps": 450, "total_steps": 750, "loss": 0.249, "lr": 4.154685949741631e-06, "epoch": 1.8, "percentage": 60.0, "elapsed_time": "0:09:26", "remaining_time": "0:06:17"}
|
| 46 |
+
{"current_steps": 460, "total_steps": 750, "loss": 0.3024, "lr": 3.926323364164684e-06, "epoch": 1.8399999999999999, "percentage": 61.33, "elapsed_time": "0:09:38", "remaining_time": "0:06:04"}
|
| 47 |
+
{"current_steps": 470, "total_steps": 750, "loss": 0.2601, "lr": 3.700286123097814e-06, "epoch": 1.88, "percentage": 62.67, "elapsed_time": "0:09:51", "remaining_time": "0:05:52"}
|
| 48 |
+
{"current_steps": 480, "total_steps": 750, "loss": 0.2632, "lr": 3.4770637728718608e-06, "epoch": 1.92, "percentage": 64.0, "elapsed_time": "0:10:03", "remaining_time": "0:05:39"}
|
| 49 |
+
{"current_steps": 490, "total_steps": 750, "loss": 0.2622, "lr": 3.2571397633909252e-06, "epoch": 1.96, "percentage": 65.33, "elapsed_time": "0:10:16", "remaining_time": "0:05:27"}
|
| 50 |
+
{"current_steps": 500, "total_steps": 750, "loss": 0.2556, "lr": 3.040990401087508e-06, "epoch": 2.0, "percentage": 66.67, "elapsed_time": "0:10:29", "remaining_time": "0:05:14"}
|
| 51 |
+
{"current_steps": 510, "total_steps": 750, "loss": 0.1551, "lr": 2.82908381734886e-06, "epoch": 2.04, "percentage": 68.0, "elapsed_time": "0:11:01", "remaining_time": "0:05:11"}
|
| 52 |
+
{"current_steps": 520, "total_steps": 750, "loss": 0.1472, "lr": 2.6218789546486235e-06, "epoch": 2.08, "percentage": 69.33, "elapsed_time": "0:11:14", "remaining_time": "0:04:58"}
|
| 53 |
+
{"current_steps": 530, "total_steps": 750, "loss": 0.1551, "lr": 2.4198245725796427e-06, "epoch": 2.12, "percentage": 70.67, "elapsed_time": "0:11:27", "remaining_time": "0:04:45"}
|
| 54 |
+
{"current_steps": 540, "total_steps": 750, "loss": 0.1369, "lr": 2.2233582759406065e-06, "epoch": 2.16, "percentage": 72.0, "elapsed_time": "0:11:39", "remaining_time": "0:04:32"}
|
| 55 |
+
{"current_steps": 550, "total_steps": 750, "loss": 0.1491, "lr": 2.0329055669814936e-06, "epoch": 2.2, "percentage": 73.33, "elapsed_time": "0:11:52", "remaining_time": "0:04:18"}
|
| 56 |
+
{"current_steps": 560, "total_steps": 750, "loss": 0.1513, "lr": 1.8488789238604676e-06, "epoch": 2.24, "percentage": 74.67, "elapsed_time": "0:12:04", "remaining_time": "0:04:05"}
|
| 57 |
+
{"current_steps": 570, "total_steps": 750, "loss": 0.1595, "lr": 1.671676907308018e-06, "epoch": 2.2800000000000002, "percentage": 76.0, "elapsed_time": "0:12:17", "remaining_time": "0:03:52"}
|
| 58 |
+
{"current_steps": 580, "total_steps": 750, "loss": 0.1425, "lr": 1.5016832974331725e-06, "epoch": 2.32, "percentage": 77.33, "elapsed_time": "0:12:29", "remaining_time": "0:03:39"}
|
| 59 |
+
{"current_steps": 590, "total_steps": 750, "loss": 0.1374, "lr": 1.339266262541249e-06, "epoch": 2.36, "percentage": 78.67, "elapsed_time": "0:12:42", "remaining_time": "0:03:26"}
|
| 60 |
+
{"current_steps": 600, "total_steps": 750, "loss": 0.1326, "lr": 1.1847775617632746e-06, "epoch": 2.4, "percentage": 80.0, "elapsed_time": "0:12:54", "remaining_time": "0:03:13"}
|
| 61 |
+
{"current_steps": 610, "total_steps": 750, "loss": 0.1321, "lr": 1.0385517832240472e-06, "epoch": 2.44, "percentage": 81.33, "elapsed_time": "0:13:07", "remaining_time": "0:03:00"}
|
| 62 |
+
{"current_steps": 620, "total_steps": 750, "loss": 0.1548, "lr": 9.009056193987569e-07, "epoch": 2.48, "percentage": 82.67, "elapsed_time": "0:13:19", "remaining_time": "0:02:47"}
|
| 63 |
+
{"current_steps": 630, "total_steps": 750, "loss": 0.1288, "lr": 7.72137181227608e-07, "epoch": 2.52, "percentage": 84.0, "elapsed_time": "0:13:32", "remaining_time": "0:02:34"}
|
| 64 |
+
{"current_steps": 640, "total_steps": 750, "loss": 0.1283, "lr": 6.52525352473905e-07, "epoch": 2.56, "percentage": 85.33, "elapsed_time": "0:13:45", "remaining_time": "0:02:21"}
|
| 65 |
+
{"current_steps": 650, "total_steps": 750, "loss": 0.1376, "lr": 5.423291857239177e-07, "epoch": 2.6, "percentage": 86.67, "elapsed_time": "0:13:57", "remaining_time": "0:02:08"}
|
| 66 |
+
{"current_steps": 660, "total_steps": 750, "loss": 0.1264, "lr": 4.417873413366702e-07, "epoch": 2.64, "percentage": 88.0, "elapsed_time": "0:14:10", "remaining_time": "0:01:56"}
|
| 67 |
+
{"current_steps": 670, "total_steps": 750, "loss": 0.1331, "lr": 3.511175705587433e-07, "epoch": 2.68, "percentage": 89.33, "elapsed_time": "0:14:23", "remaining_time": "0:01:43"}
|
| 68 |
+
{"current_steps": 680, "total_steps": 750, "loss": 0.1232, "lr": 2.705162439235648e-07, "epoch": 2.7199999999999998, "percentage": 90.67, "elapsed_time": "0:14:36", "remaining_time": "0:01:30"}
|
| 69 |
+
{"current_steps": 690, "total_steps": 750, "loss": 0.1122, "lr": 2.0015792595656225e-07, "epoch": 2.76, "percentage": 92.0, "elapsed_time": "0:14:48", "remaining_time": "0:01:17"}
|
| 70 |
+
{"current_steps": 700, "total_steps": 750, "loss": 0.1162, "lr": 1.4019499710726913e-07, "epoch": 2.8, "percentage": 93.33, "elapsed_time": "0:15:01", "remaining_time": "0:01:04"}
|
| 71 |
+
{"current_steps": 710, "total_steps": 750, "loss": 0.1209, "lr": 9.075732372720414e-08, "epoch": 2.84, "percentage": 94.67, "elapsed_time": "0:15:13", "remaining_time": "0:00:51"}
|
| 72 |
+
{"current_steps": 720, "total_steps": 750, "loss": 0.1304, "lr": 5.19519768082738e-08, "epoch": 2.88, "percentage": 96.0, "elapsed_time": "0:15:26", "remaining_time": "0:00:38"}
|
| 73 |
+
{"current_steps": 730, "total_steps": 750, "loss": 0.1373, "lr": 2.386300009084408e-08, "epoch": 2.92, "percentage": 97.33, "elapsed_time": "0:15:39", "remaining_time": "0:00:25"}
|
| 74 |
+
{"current_steps": 740, "total_steps": 750, "loss": 0.1262, "lr": 6.551228043715218e-09, "epoch": 2.96, "percentage": 98.67, "elapsed_time": "0:15:51", "remaining_time": "0:00:12"}
|
| 75 |
+
{"current_steps": 750, "total_steps": 750, "loss": 0.1221, "lr": 5.4154110206150465e-11, "epoch": 3.0, "percentage": 100.0, "elapsed_time": "0:16:04", "remaining_time": "0:00:00"}
|
| 76 |
+
{"current_steps": 750, "total_steps": 750, "epoch": 3.0, "percentage": 100.0, "elapsed_time": "0:16:24", "remaining_time": "0:00:00"}
|
trainer_state.json
ADDED
|
@@ -0,0 +1,568 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_global_step": null,
|
| 3 |
+
"best_metric": null,
|
| 4 |
+
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 3.0,
|
| 6 |
+
"eval_steps": 500,
|
| 7 |
+
"global_step": 750,
|
| 8 |
+
"is_hyper_param_search": false,
|
| 9 |
+
"is_local_process_zero": true,
|
| 10 |
+
"is_world_process_zero": true,
|
| 11 |
+
"log_history": [
|
| 12 |
+
{
|
| 13 |
+
"epoch": 0.04,
|
| 14 |
+
"grad_norm": 14.964976108186999,
|
| 15 |
+
"learning_rate": 1.2000000000000002e-06,
|
| 16 |
+
"loss": 1.0456,
|
| 17 |
+
"step": 10
|
| 18 |
+
},
|
| 19 |
+
{
|
| 20 |
+
"epoch": 0.08,
|
| 21 |
+
"grad_norm": 13.248201001919654,
|
| 22 |
+
"learning_rate": 2.5333333333333338e-06,
|
| 23 |
+
"loss": 0.901,
|
| 24 |
+
"step": 20
|
| 25 |
+
},
|
| 26 |
+
{
|
| 27 |
+
"epoch": 0.12,
|
| 28 |
+
"grad_norm": 11.343520888209307,
|
| 29 |
+
"learning_rate": 3.866666666666667e-06,
|
| 30 |
+
"loss": 0.743,
|
| 31 |
+
"step": 30
|
| 32 |
+
},
|
| 33 |
+
{
|
| 34 |
+
"epoch": 0.16,
|
| 35 |
+
"grad_norm": 10.573065944855244,
|
| 36 |
+
"learning_rate": 5.2e-06,
|
| 37 |
+
"loss": 0.5085,
|
| 38 |
+
"step": 40
|
| 39 |
+
},
|
| 40 |
+
{
|
| 41 |
+
"epoch": 0.2,
|
| 42 |
+
"grad_norm": 7.449411452156014,
|
| 43 |
+
"learning_rate": 6.533333333333334e-06,
|
| 44 |
+
"loss": 0.5229,
|
| 45 |
+
"step": 50
|
| 46 |
+
},
|
| 47 |
+
{
|
| 48 |
+
"epoch": 0.24,
|
| 49 |
+
"grad_norm": 6.113028114593532,
|
| 50 |
+
"learning_rate": 7.866666666666667e-06,
|
| 51 |
+
"loss": 0.4071,
|
| 52 |
+
"step": 60
|
| 53 |
+
},
|
| 54 |
+
{
|
| 55 |
+
"epoch": 0.28,
|
| 56 |
+
"grad_norm": 6.458832687302615,
|
| 57 |
+
"learning_rate": 9.200000000000002e-06,
|
| 58 |
+
"loss": 0.3846,
|
| 59 |
+
"step": 70
|
| 60 |
+
},
|
| 61 |
+
{
|
| 62 |
+
"epoch": 0.32,
|
| 63 |
+
"grad_norm": 9.014495669830236,
|
| 64 |
+
"learning_rate": 9.99913355769784e-06,
|
| 65 |
+
"loss": 0.4199,
|
| 66 |
+
"step": 80
|
| 67 |
+
},
|
| 68 |
+
{
|
| 69 |
+
"epoch": 0.36,
|
| 70 |
+
"grad_norm": 6.7935996831924115,
|
| 71 |
+
"learning_rate": 9.989389530100242e-06,
|
| 72 |
+
"loss": 0.4515,
|
| 73 |
+
"step": 90
|
| 74 |
+
},
|
| 75 |
+
{
|
| 76 |
+
"epoch": 0.4,
|
| 77 |
+
"grad_norm": 8.41349667436057,
|
| 78 |
+
"learning_rate": 9.968839595802982e-06,
|
| 79 |
+
"loss": 0.4596,
|
| 80 |
+
"step": 100
|
| 81 |
+
},
|
| 82 |
+
{
|
| 83 |
+
"epoch": 0.44,
|
| 84 |
+
"grad_norm": 5.730189913709734,
|
| 85 |
+
"learning_rate": 9.937528261387753e-06,
|
| 86 |
+
"loss": 0.4392,
|
| 87 |
+
"step": 110
|
| 88 |
+
},
|
| 89 |
+
{
|
| 90 |
+
"epoch": 0.48,
|
| 91 |
+
"grad_norm": 8.608526958184274,
|
| 92 |
+
"learning_rate": 9.89552334023258e-06,
|
| 93 |
+
"loss": 0.4601,
|
| 94 |
+
"step": 120
|
| 95 |
+
},
|
| 96 |
+
{
|
| 97 |
+
"epoch": 0.52,
|
| 98 |
+
"grad_norm": 10.626765774764173,
|
| 99 |
+
"learning_rate": 9.842915805643156e-06,
|
| 100 |
+
"loss": 0.4504,
|
| 101 |
+
"step": 130
|
| 102 |
+
},
|
| 103 |
+
{
|
| 104 |
+
"epoch": 0.56,
|
| 105 |
+
"grad_norm": 6.044435310014289,
|
| 106 |
+
"learning_rate": 9.779819593824909e-06,
|
| 107 |
+
"loss": 0.4383,
|
| 108 |
+
"step": 140
|
| 109 |
+
},
|
| 110 |
+
{
|
| 111 |
+
"epoch": 0.6,
|
| 112 |
+
"grad_norm": 8.95504802835063,
|
| 113 |
+
"learning_rate": 9.70637135712256e-06,
|
| 114 |
+
"loss": 0.4168,
|
| 115 |
+
"step": 150
|
| 116 |
+
},
|
| 117 |
+
{
|
| 118 |
+
"epoch": 0.64,
|
| 119 |
+
"grad_norm": 6.062300766541639,
|
| 120 |
+
"learning_rate": 9.622730168061568e-06,
|
| 121 |
+
"loss": 0.4674,
|
| 122 |
+
"step": 160
|
| 123 |
+
},
|
| 124 |
+
{
|
| 125 |
+
"epoch": 0.68,
|
| 126 |
+
"grad_norm": 5.914652504110795,
|
| 127 |
+
"learning_rate": 9.529077174832466e-06,
|
| 128 |
+
"loss": 0.4282,
|
| 129 |
+
"step": 170
|
| 130 |
+
},
|
| 131 |
+
{
|
| 132 |
+
"epoch": 0.72,
|
| 133 |
+
"grad_norm": 6.998871084061125,
|
| 134 |
+
"learning_rate": 9.425615208964217e-06,
|
| 135 |
+
"loss": 0.4303,
|
| 136 |
+
"step": 180
|
| 137 |
+
},
|
| 138 |
+
{
|
| 139 |
+
"epoch": 0.76,
|
| 140 |
+
"grad_norm": 7.620254502582685,
|
| 141 |
+
"learning_rate": 9.312568346036288e-06,
|
| 142 |
+
"loss": 0.429,
|
| 143 |
+
"step": 190
|
| 144 |
+
},
|
| 145 |
+
{
|
| 146 |
+
"epoch": 0.8,
|
| 147 |
+
"grad_norm": 5.8313093153091105,
|
| 148 |
+
"learning_rate": 9.190181420380838e-06,
|
| 149 |
+
"loss": 0.486,
|
| 150 |
+
"step": 200
|
| 151 |
+
},
|
| 152 |
+
{
|
| 153 |
+
"epoch": 0.84,
|
| 154 |
+
"grad_norm": 7.893697074335927,
|
| 155 |
+
"learning_rate": 9.058719494826076e-06,
|
| 156 |
+
"loss": 0.3998,
|
| 157 |
+
"step": 210
|
| 158 |
+
},
|
| 159 |
+
{
|
| 160 |
+
"epoch": 0.88,
|
| 161 |
+
"grad_norm": 7.997975976474495,
|
| 162 |
+
"learning_rate": 8.9184672866292e-06,
|
| 163 |
+
"loss": 0.4351,
|
| 164 |
+
"step": 220
|
| 165 |
+
},
|
| 166 |
+
{
|
| 167 |
+
"epoch": 0.92,
|
| 168 |
+
"grad_norm": 6.112788775519066,
|
| 169 |
+
"learning_rate": 8.769728550842217e-06,
|
| 170 |
+
"loss": 0.4512,
|
| 171 |
+
"step": 230
|
| 172 |
+
},
|
| 173 |
+
{
|
| 174 |
+
"epoch": 0.96,
|
| 175 |
+
"grad_norm": 5.94627559545741,
|
| 176 |
+
"learning_rate": 8.61282542244614e-06,
|
| 177 |
+
"loss": 0.421,
|
| 178 |
+
"step": 240
|
| 179 |
+
},
|
| 180 |
+
{
|
| 181 |
+
"epoch": 1.0,
|
| 182 |
+
"grad_norm": 6.918961512851811,
|
| 183 |
+
"learning_rate": 8.44809771867835e-06,
|
| 184 |
+
"loss": 0.3976,
|
| 185 |
+
"step": 250
|
| 186 |
+
},
|
| 187 |
+
{
|
| 188 |
+
"epoch": 1.04,
|
| 189 |
+
"grad_norm": 5.214035260048491,
|
| 190 |
+
"learning_rate": 8.275902203064125e-06,
|
| 191 |
+
"loss": 0.2612,
|
| 192 |
+
"step": 260
|
| 193 |
+
},
|
| 194 |
+
{
|
| 195 |
+
"epoch": 1.08,
|
| 196 |
+
"grad_norm": 11.428160470655266,
|
| 197 |
+
"learning_rate": 8.096611812746302e-06,
|
| 198 |
+
"loss": 0.3022,
|
| 199 |
+
"step": 270
|
| 200 |
+
},
|
| 201 |
+
{
|
| 202 |
+
"epoch": 1.12,
|
| 203 |
+
"grad_norm": 7.610014947005798,
|
| 204 |
+
"learning_rate": 7.910614850786448e-06,
|
| 205 |
+
"loss": 0.2954,
|
| 206 |
+
"step": 280
|
| 207 |
+
},
|
| 208 |
+
{
|
| 209 |
+
"epoch": 1.16,
|
| 210 |
+
"grad_norm": 4.90410268397758,
|
| 211 |
+
"learning_rate": 7.718314145186918e-06,
|
| 212 |
+
"loss": 0.331,
|
| 213 |
+
"step": 290
|
| 214 |
+
},
|
| 215 |
+
{
|
| 216 |
+
"epoch": 1.2,
|
| 217 |
+
"grad_norm": 6.252190194552348,
|
| 218 |
+
"learning_rate": 7.520126176455084e-06,
|
| 219 |
+
"loss": 0.2971,
|
| 220 |
+
"step": 300
|
| 221 |
+
},
|
| 222 |
+
{
|
| 223 |
+
"epoch": 1.24,
|
| 224 |
+
"grad_norm": 4.17312381107693,
|
| 225 |
+
"learning_rate": 7.31648017559931e-06,
|
| 226 |
+
"loss": 0.2885,
|
| 227 |
+
"step": 310
|
| 228 |
+
},
|
| 229 |
+
{
|
| 230 |
+
"epoch": 1.28,
|
| 231 |
+
"grad_norm": 5.26853759901534,
|
| 232 |
+
"learning_rate": 7.107817194510157e-06,
|
| 233 |
+
"loss": 0.2848,
|
| 234 |
+
"step": 320
|
| 235 |
+
},
|
| 236 |
+
{
|
| 237 |
+
"epoch": 1.32,
|
| 238 |
+
"grad_norm": 6.485801444034796,
|
| 239 |
+
"learning_rate": 6.8945891507402075e-06,
|
| 240 |
+
"loss": 0.2989,
|
| 241 |
+
"step": 330
|
| 242 |
+
},
|
| 243 |
+
{
|
| 244 |
+
"epoch": 1.3599999999999999,
|
| 245 |
+
"grad_norm": 5.972772954729457,
|
| 246 |
+
"learning_rate": 6.677257848751276e-06,
|
| 247 |
+
"loss": 0.2939,
|
| 248 |
+
"step": 340
|
| 249 |
+
},
|
| 250 |
+
{
|
| 251 |
+
"epoch": 1.4,
|
| 252 |
+
"grad_norm": 5.935090104641661,
|
| 253 |
+
"learning_rate": 6.456293979748778e-06,
|
| 254 |
+
"loss": 0.2898,
|
| 255 |
+
"step": 350
|
| 256 |
+
},
|
| 257 |
+
{
|
| 258 |
+
"epoch": 1.44,
|
| 259 |
+
"grad_norm": 7.019147334619332,
|
| 260 |
+
"learning_rate": 6.23217610226939e-06,
|
| 261 |
+
"loss": 0.2682,
|
| 262 |
+
"step": 360
|
| 263 |
+
},
|
| 264 |
+
{
|
| 265 |
+
"epoch": 1.48,
|
| 266 |
+
"grad_norm": 6.194674693800203,
|
| 267 |
+
"learning_rate": 6.005389605729824e-06,
|
| 268 |
+
"loss": 0.2777,
|
| 269 |
+
"step": 370
|
| 270 |
+
},
|
| 271 |
+
{
|
| 272 |
+
"epoch": 1.52,
|
| 273 |
+
"grad_norm": 5.012869958485124,
|
| 274 |
+
"learning_rate": 5.776425659181438e-06,
|
| 275 |
+
"loss": 0.268,
|
| 276 |
+
"step": 380
|
| 277 |
+
},
|
| 278 |
+
{
|
| 279 |
+
"epoch": 1.56,
|
| 280 |
+
"grad_norm": 6.186940301764657,
|
| 281 |
+
"learning_rate": 5.54578014754744e-06,
|
| 282 |
+
"loss": 0.2945,
|
| 283 |
+
"step": 390
|
| 284 |
+
},
|
| 285 |
+
{
|
| 286 |
+
"epoch": 1.6,
|
| 287 |
+
"grad_norm": 5.959462554843478,
|
| 288 |
+
"learning_rate": 5.3139525976465675e-06,
|
| 289 |
+
"loss": 0.3164,
|
| 290 |
+
"step": 400
|
| 291 |
+
},
|
| 292 |
+
{
|
| 293 |
+
"epoch": 1.6400000000000001,
|
| 294 |
+
"grad_norm": 5.347202908640427,
|
| 295 |
+
"learning_rate": 5.081445096329229e-06,
|
| 296 |
+
"loss": 0.262,
|
| 297 |
+
"step": 410
|
| 298 |
+
},
|
| 299 |
+
{
|
| 300 |
+
"epoch": 1.6800000000000002,
|
| 301 |
+
"grad_norm": 6.356975909974464,
|
| 302 |
+
"learning_rate": 4.8487612030691975e-06,
|
| 303 |
+
"loss": 0.2652,
|
| 304 |
+
"step": 420
|
| 305 |
+
},
|
| 306 |
+
{
|
| 307 |
+
"epoch": 1.72,
|
| 308 |
+
"grad_norm": 5.356416954731608,
|
| 309 |
+
"learning_rate": 4.6164048593659076e-06,
|
| 310 |
+
"loss": 0.2418,
|
| 311 |
+
"step": 430
|
| 312 |
+
},
|
| 313 |
+
{
|
| 314 |
+
"epoch": 1.76,
|
| 315 |
+
"grad_norm": 4.741200353283629,
|
| 316 |
+
"learning_rate": 4.384879297319398e-06,
|
| 317 |
+
"loss": 0.2714,
|
| 318 |
+
"step": 440
|
| 319 |
+
},
|
| 320 |
+
{
|
| 321 |
+
"epoch": 1.8,
|
| 322 |
+
"grad_norm": 4.5775651416127925,
|
| 323 |
+
"learning_rate": 4.154685949741631e-06,
|
| 324 |
+
"loss": 0.249,
|
| 325 |
+
"step": 450
|
| 326 |
+
},
|
| 327 |
+
{
|
| 328 |
+
"epoch": 1.8399999999999999,
|
| 329 |
+
"grad_norm": 4.6524509736470065,
|
| 330 |
+
"learning_rate": 3.926323364164684e-06,
|
| 331 |
+
"loss": 0.3024,
|
| 332 |
+
"step": 460
|
| 333 |
+
},
|
| 334 |
+
{
|
| 335 |
+
"epoch": 1.88,
|
| 336 |
+
"grad_norm": 4.870591389397286,
|
| 337 |
+
"learning_rate": 3.700286123097814e-06,
|
| 338 |
+
"loss": 0.2601,
|
| 339 |
+
"step": 470
|
| 340 |
+
},
|
| 341 |
+
{
|
| 342 |
+
"epoch": 1.92,
|
| 343 |
+
"grad_norm": 6.825736988205842,
|
| 344 |
+
"learning_rate": 3.4770637728718608e-06,
|
| 345 |
+
"loss": 0.2632,
|
| 346 |
+
"step": 480
|
| 347 |
+
},
|
| 348 |
+
{
|
| 349 |
+
"epoch": 1.96,
|
| 350 |
+
"grad_norm": 5.5580038236654,
|
| 351 |
+
"learning_rate": 3.2571397633909252e-06,
|
| 352 |
+
"loss": 0.2622,
|
| 353 |
+
"step": 490
|
| 354 |
+
},
|
| 355 |
+
{
|
| 356 |
+
"epoch": 2.0,
|
| 357 |
+
"grad_norm": 5.146632505710646,
|
| 358 |
+
"learning_rate": 3.040990401087508e-06,
|
| 359 |
+
"loss": 0.2556,
|
| 360 |
+
"step": 500
|
| 361 |
+
},
|
| 362 |
+
{
|
| 363 |
+
"epoch": 2.04,
|
| 364 |
+
"grad_norm": 3.693695132405,
|
| 365 |
+
"learning_rate": 2.82908381734886e-06,
|
| 366 |
+
"loss": 0.1551,
|
| 367 |
+
"step": 510
|
| 368 |
+
},
|
| 369 |
+
{
|
| 370 |
+
"epoch": 2.08,
|
| 371 |
+
"grad_norm": 3.5139004587947196,
|
| 372 |
+
"learning_rate": 2.6218789546486235e-06,
|
| 373 |
+
"loss": 0.1472,
|
| 374 |
+
"step": 520
|
| 375 |
+
},
|
| 376 |
+
{
|
| 377 |
+
"epoch": 2.12,
|
| 378 |
+
"grad_norm": 2.4427933540261515,
|
| 379 |
+
"learning_rate": 2.4198245725796427e-06,
|
| 380 |
+
"loss": 0.1551,
|
| 381 |
+
"step": 530
|
| 382 |
+
},
|
| 383 |
+
{
|
| 384 |
+
"epoch": 2.16,
|
| 385 |
+
"grad_norm": 3.5327689862898546,
|
| 386 |
+
"learning_rate": 2.2233582759406065e-06,
|
| 387 |
+
"loss": 0.1369,
|
| 388 |
+
"step": 540
|
| 389 |
+
},
|
| 390 |
+
{
|
| 391 |
+
"epoch": 2.2,
|
| 392 |
+
"grad_norm": 4.368671164271065,
|
| 393 |
+
"learning_rate": 2.0329055669814936e-06,
|
| 394 |
+
"loss": 0.1491,
|
| 395 |
+
"step": 550
|
| 396 |
+
},
|
| 397 |
+
{
|
| 398 |
+
"epoch": 2.24,
|
| 399 |
+
"grad_norm": 4.844760203284172,
|
| 400 |
+
"learning_rate": 1.8488789238604676e-06,
|
| 401 |
+
"loss": 0.1513,
|
| 402 |
+
"step": 560
|
| 403 |
+
},
|
| 404 |
+
{
|
| 405 |
+
"epoch": 2.2800000000000002,
|
| 406 |
+
"grad_norm": 5.389592493961255,
|
| 407 |
+
"learning_rate": 1.671676907308018e-06,
|
| 408 |
+
"loss": 0.1595,
|
| 409 |
+
"step": 570
|
| 410 |
+
},
|
| 411 |
+
{
|
| 412 |
+
"epoch": 2.32,
|
| 413 |
+
"grad_norm": 6.080906290251134,
|
| 414 |
+
"learning_rate": 1.5016832974331725e-06,
|
| 415 |
+
"loss": 0.1425,
|
| 416 |
+
"step": 580
|
| 417 |
+
},
|
| 418 |
+
{
|
| 419 |
+
"epoch": 2.36,
|
| 420 |
+
"grad_norm": 3.7228796010958787,
|
| 421 |
+
"learning_rate": 1.339266262541249e-06,
|
| 422 |
+
"loss": 0.1374,
|
| 423 |
+
"step": 590
|
| 424 |
+
},
|
| 425 |
+
{
|
| 426 |
+
"epoch": 2.4,
|
| 427 |
+
"grad_norm": 4.45024442529631,
|
| 428 |
+
"learning_rate": 1.1847775617632746e-06,
|
| 429 |
+
"loss": 0.1326,
|
| 430 |
+
"step": 600
|
| 431 |
+
},
|
| 432 |
+
{
|
| 433 |
+
"epoch": 2.44,
|
| 434 |
+
"grad_norm": 5.115769379241484,
|
| 435 |
+
"learning_rate": 1.0385517832240472e-06,
|
| 436 |
+
"loss": 0.1321,
|
| 437 |
+
"step": 610
|
| 438 |
+
},
|
| 439 |
+
{
|
| 440 |
+
"epoch": 2.48,
|
| 441 |
+
"grad_norm": 5.050238431161297,
|
| 442 |
+
"learning_rate": 9.009056193987569e-07,
|
| 443 |
+
"loss": 0.1548,
|
| 444 |
+
"step": 620
|
| 445 |
+
},
|
| 446 |
+
{
|
| 447 |
+
"epoch": 2.52,
|
| 448 |
+
"grad_norm": 6.625572108695121,
|
| 449 |
+
"learning_rate": 7.72137181227608e-07,
|
| 450 |
+
"loss": 0.1288,
|
| 451 |
+
"step": 630
|
| 452 |
+
},
|
| 453 |
+
{
|
| 454 |
+
"epoch": 2.56,
|
| 455 |
+
"grad_norm": 4.091188373097037,
|
| 456 |
+
"learning_rate": 6.52525352473905e-07,
|
| 457 |
+
"loss": 0.1283,
|
| 458 |
+
"step": 640
|
| 459 |
+
},
|
| 460 |
+
{
|
| 461 |
+
"epoch": 2.6,
|
| 462 |
+
"grad_norm": 5.9420896735395,
|
| 463 |
+
"learning_rate": 5.423291857239177e-07,
|
| 464 |
+
"loss": 0.1376,
|
| 465 |
+
"step": 650
|
| 466 |
+
},
|
| 467 |
+
{
|
| 468 |
+
"epoch": 2.64,
|
| 469 |
+
"grad_norm": 5.768293093992854,
|
| 470 |
+
"learning_rate": 4.417873413366702e-07,
|
| 471 |
+
"loss": 0.1264,
|
| 472 |
+
"step": 660
|
| 473 |
+
},
|
| 474 |
+
{
|
| 475 |
+
"epoch": 2.68,
|
| 476 |
+
"grad_norm": 5.704351644337552,
|
| 477 |
+
"learning_rate": 3.511175705587433e-07,
|
| 478 |
+
"loss": 0.1331,
|
| 479 |
+
"step": 670
|
| 480 |
+
},
|
| 481 |
+
{
|
| 482 |
+
"epoch": 2.7199999999999998,
|
| 483 |
+
"grad_norm": 4.685177109012829,
|
| 484 |
+
"learning_rate": 2.705162439235648e-07,
|
| 485 |
+
"loss": 0.1232,
|
| 486 |
+
"step": 680
|
| 487 |
+
},
|
| 488 |
+
{
|
| 489 |
+
"epoch": 2.76,
|
| 490 |
+
"grad_norm": 3.406413777853003,
|
| 491 |
+
"learning_rate": 2.0015792595656225e-07,
|
| 492 |
+
"loss": 0.1122,
|
| 493 |
+
"step": 690
|
| 494 |
+
},
|
| 495 |
+
{
|
| 496 |
+
"epoch": 2.8,
|
| 497 |
+
"grad_norm": 5.602707964005007,
|
| 498 |
+
"learning_rate": 1.4019499710726913e-07,
|
| 499 |
+
"loss": 0.1162,
|
| 500 |
+
"step": 700
|
| 501 |
+
},
|
| 502 |
+
{
|
| 503 |
+
"epoch": 2.84,
|
| 504 |
+
"grad_norm": 3.674464349451227,
|
| 505 |
+
"learning_rate": 9.075732372720414e-08,
|
| 506 |
+
"loss": 0.1209,
|
| 507 |
+
"step": 710
|
| 508 |
+
},
|
| 509 |
+
{
|
| 510 |
+
"epoch": 2.88,
|
| 511 |
+
"grad_norm": 4.922409104144764,
|
| 512 |
+
"learning_rate": 5.19519768082738e-08,
|
| 513 |
+
"loss": 0.1304,
|
| 514 |
+
"step": 720
|
| 515 |
+
},
|
| 516 |
+
{
|
| 517 |
+
"epoch": 2.92,
|
| 518 |
+
"grad_norm": 4.9538201554652375,
|
| 519 |
+
"learning_rate": 2.386300009084408e-08,
|
| 520 |
+
"loss": 0.1373,
|
| 521 |
+
"step": 730
|
| 522 |
+
},
|
| 523 |
+
{
|
| 524 |
+
"epoch": 2.96,
|
| 525 |
+
"grad_norm": 7.024299816771905,
|
| 526 |
+
"learning_rate": 6.551228043715218e-09,
|
| 527 |
+
"loss": 0.1262,
|
| 528 |
+
"step": 740
|
| 529 |
+
},
|
| 530 |
+
{
|
| 531 |
+
"epoch": 3.0,
|
| 532 |
+
"grad_norm": 4.968701193243566,
|
| 533 |
+
"learning_rate": 5.4154110206150465e-11,
|
| 534 |
+
"loss": 0.1221,
|
| 535 |
+
"step": 750
|
| 536 |
+
},
|
| 537 |
+
{
|
| 538 |
+
"epoch": 3.0,
|
| 539 |
+
"step": 750,
|
| 540 |
+
"total_flos": 5335736082432.0,
|
| 541 |
+
"train_loss": 0.30388372325897217,
|
| 542 |
+
"train_runtime": 989.6325,
|
| 543 |
+
"train_samples_per_second": 3.031,
|
| 544 |
+
"train_steps_per_second": 0.758
|
| 545 |
+
}
|
| 546 |
+
],
|
| 547 |
+
"logging_steps": 10,
|
| 548 |
+
"max_steps": 750,
|
| 549 |
+
"num_input_tokens_seen": 0,
|
| 550 |
+
"num_train_epochs": 3,
|
| 551 |
+
"save_steps": 500,
|
| 552 |
+
"stateful_callbacks": {
|
| 553 |
+
"TrainerControl": {
|
| 554 |
+
"args": {
|
| 555 |
+
"should_epoch_stop": false,
|
| 556 |
+
"should_evaluate": false,
|
| 557 |
+
"should_log": false,
|
| 558 |
+
"should_save": true,
|
| 559 |
+
"should_training_stop": true
|
| 560 |
+
},
|
| 561 |
+
"attributes": {}
|
| 562 |
+
}
|
| 563 |
+
},
|
| 564 |
+
"total_flos": 5335736082432.0,
|
| 565 |
+
"train_batch_size": 1,
|
| 566 |
+
"trial_name": null,
|
| 567 |
+
"trial_params": null
|
| 568 |
+
}
|
training_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6121c699a147321f7f7c9b44f2b30592b65183fc798429692c67a28937a8947a
|
| 3 |
+
size 8145
|
training_loss.png
ADDED
|
vocab.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|