初始化项目，由ModelHub XC社区提供模型
Model: tzchen07/SG_X9e Source: Original Platform
This commit is contained in:
ModelHub XC
29
checkpoint-498/chat_template.jinja
Normal file
29
checkpoint-498/chat_template.jinja
Normal file
@@ -0,0 +1,29 @@
|
||||
{%- if not guideline and messages[0].role == "system" %}
|
||||
{%- set guideline = messages[0].content %}
|
||||
{%- set messages = messages[1:] %}
|
||||
{%- endif %}
|
||||
|
||||
{{- bos_token }}
|
||||
{%- if messages[-1].role == 'user' %}
|
||||
{{- "You are a policy expert trying to help determine whether a user prompt is in violation of the defined safety policies.\n\n" }}
|
||||
{{- "<start_of_turn>\n" }}
|
||||
{{- "Human Question: " + messages[-1].content }}
|
||||
{{- "\n<end_of_turn>\n\n" }}
|
||||
{{- "Our safety principle is defined in the below:\n\n" }}
|
||||
{{- "* " + guideline + "\n" }}
|
||||
{{- "\n===\n\n" }}
|
||||
{{- "Does the human question violate the above principle? Your answer must start with 'Yes' or 'No'. And then walk through step by step to be sure we answer correctly.\n\n" }}
|
||||
{%- else %}
|
||||
{{- "You are a policy expert trying to help determine whether an AI response to a prompt is in violation of the defined safety policies.\n\n" }}
|
||||
{{- "<start_of_turn>\n" }}
|
||||
{{- "Human Question: " + messages[-2].content }}
|
||||
{{- "\n<end_of_turn>\n" }}
|
||||
{{- "<start_of_turn>\n" }}
|
||||
{{- "Chatbot Response: " + messages[-1].content }}
|
||||
{{- "\n<end_of_turn>\n\n" }}
|
||||
{{- "Our safety principle is defined in the below:\n\n" }}
|
||||
{{- "* " + guideline + "\n" }}
|
||||
{{- "\n===\n\n" }}
|
||||
{{- "Does the Chatbot Response violate the above principle? Your answer must start with 'Yes' or 'No'. And then walk through step by step to be sure we answer correctly.\n\n" }}
|
||||
{%- endif %}
|
||||
|
||||
60
checkpoint-498/config.json
Normal file
60
checkpoint-498/config.json
Normal file
@@ -0,0 +1,60 @@
|
||||
{
|
||||
"architectures": [
|
||||
"Gemma2ForCausalLM"
|
||||
],
|
||||
"attention_bias": false,
|
||||
"attention_dropout": 0.0,
|
||||
"attn_logit_softcapping": 50.0,
|
||||
"bos_token_id": 2,
|
||||
"cache_implementation": "hybrid",
|
||||
"dtype": "float32",
|
||||
"eos_token_id": 107,
|
||||
"final_logit_softcapping": 30.0,
|
||||
"head_dim": 256,
|
||||
"hidden_act": "gelu_pytorch_tanh",
|
||||
"hidden_activation": "gelu_pytorch_tanh",
|
||||
"hidden_size": 2304,
|
||||
"initializer_range": 0.02,
|
||||
"intermediate_size": 9216,
|
||||
"layer_types": [
|
||||
"sliding_attention",
|
||||
"full_attention",
|
||||
"sliding_attention",
|
||||
"full_attention",
|
||||
"sliding_attention",
|
||||
"full_attention",
|
||||
"sliding_attention",
|
||||
"full_attention",
|
||||
"sliding_attention",
|
||||
"full_attention",
|
||||
"sliding_attention",
|
||||
"full_attention",
|
||||
"sliding_attention",
|
||||
"full_attention",
|
||||
"sliding_attention",
|
||||
"full_attention",
|
||||
"sliding_attention",
|
||||
"full_attention",
|
||||
"sliding_attention",
|
||||
"full_attention",
|
||||
"sliding_attention",
|
||||
"full_attention",
|
||||
"sliding_attention",
|
||||
"full_attention",
|
||||
"sliding_attention",
|
||||
"full_attention"
|
||||
],
|
||||
"max_position_embeddings": 8192,
|
||||
"model_type": "gemma2",
|
||||
"num_attention_heads": 8,
|
||||
"num_hidden_layers": 26,
|
||||
"num_key_value_heads": 4,
|
||||
"pad_token_id": 0,
|
||||
"query_pre_attn_scalar": 224,
|
||||
"rms_norm_eps": 1e-06,
|
||||
"rope_theta": 10000.0,
|
||||
"sliding_window": 4096,
|
||||
"transformers_version": "4.57.1",
|
||||
"use_cache": false,
|
||||
"vocab_size": 256000
|
||||
}
|
||||
11
checkpoint-498/generation_config.json
Normal file
11
checkpoint-498/generation_config.json
Normal file
@@ -0,0 +1,11 @@
|
||||
{
|
||||
"_from_model_config": true,
|
||||
"bos_token_id": 2,
|
||||
"cache_implementation": "hybrid",
|
||||
"eos_token_id": [
|
||||
107,
|
||||
1
|
||||
],
|
||||
"pad_token_id": 0,
|
||||
"transformers_version": "4.57.1"
|
||||
}
|
||||
3
checkpoint-498/model-00001-of-00003.safetensors
Normal file
3
checkpoint-498/model-00001-of-00003.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:27f3efcf7fa387005a2447be0b4ae55681e84d5d0385c0be339b8e43b61ecd58
|
||||
size 4992576136
|
||||
3
checkpoint-498/model-00002-of-00003.safetensors
Normal file
3
checkpoint-498/model-00002-of-00003.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:f49c8916c976384e6cf23c4dfebbf04618e1741ca8838c69a4b3b55267b3b65c
|
||||
size 4983443424
|
||||
3
checkpoint-498/model-00003-of-00003.safetensors
Normal file
3
checkpoint-498/model-00003-of-00003.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:bcb13e57128059be2ef3f6db700237dec8ad73862fdcf2665b0cc904b48b23c0
|
||||
size 481381384
|
||||
296
checkpoint-498/model.safetensors.index.json
Normal file
296
checkpoint-498/model.safetensors.index.json
Normal file
@@ -0,0 +1,296 @@
|
||||
{
|
||||
"metadata": {
|
||||
"total_parameters": 2614341888,
|
||||
"total_size": 10457367552
|
||||
},
|
||||
"weight_map": {
|
||||
"model.embed_tokens.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.0.input_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.0.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.0.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.0.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.0.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.0.post_feedforward_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.0.pre_feedforward_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.0.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.0.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.0.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.0.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.1.input_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.1.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.1.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.1.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.1.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.1.post_feedforward_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.1.pre_feedforward_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.1.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.1.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.1.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.1.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.10.input_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.10.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.10.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.10.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.10.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.10.post_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.10.pre_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.10.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.10.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.10.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.10.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.11.input_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.11.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.11.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.11.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.11.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.11.post_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.11.pre_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.11.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.11.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.11.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.11.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.12.input_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.12.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.12.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.12.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.12.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.12.post_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.12.pre_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.12.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.12.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.12.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.12.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.13.input_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.13.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.13.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.13.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.13.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.13.post_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.13.pre_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.13.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.13.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.13.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.13.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.14.input_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.14.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.14.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.14.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.14.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.14.post_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.14.pre_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.14.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.14.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.14.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.14.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.15.input_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.15.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.15.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.15.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.15.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.15.post_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.15.pre_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.15.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.15.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.15.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.15.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.16.input_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.16.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.16.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.16.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.16.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.16.post_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.16.pre_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.16.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.16.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.16.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.16.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.17.input_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.17.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.17.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.17.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.17.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.17.post_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.17.pre_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.17.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.17.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.17.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.17.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.18.input_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.18.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.18.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.18.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.18.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.18.post_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.18.pre_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.18.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.18.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.18.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.18.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.19.input_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.19.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.19.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.19.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.19.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.19.post_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.19.pre_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.19.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.19.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.19.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.19.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.2.input_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.2.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.2.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.2.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.2.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.2.post_feedforward_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.2.pre_feedforward_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.2.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.2.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.2.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.2.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.20.input_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.20.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.20.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.20.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.20.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.20.post_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.20.pre_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.20.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.20.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.20.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.20.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.21.input_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.21.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.21.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.21.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.21.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.21.post_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.21.pre_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.21.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.21.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.21.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.21.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.22.input_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.22.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.22.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.22.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.22.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.22.post_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.22.pre_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.22.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.22.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.22.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.22.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.23.input_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.23.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.23.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.23.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.23.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.23.post_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.23.pre_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.23.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.23.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.23.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.23.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.24.input_layernorm.weight": "model-00003-of-00003.safetensors",
|
||||
"model.layers.24.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
|
||||
"model.layers.24.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.24.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
|
||||
"model.layers.24.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
|
||||
"model.layers.24.post_feedforward_layernorm.weight": "model-00003-of-00003.safetensors",
|
||||
"model.layers.24.pre_feedforward_layernorm.weight": "model-00003-of-00003.safetensors",
|
||||
"model.layers.24.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.24.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.24.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.24.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.25.input_layernorm.weight": "model-00003-of-00003.safetensors",
|
||||
"model.layers.25.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
|
||||
"model.layers.25.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
|
||||
"model.layers.25.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
|
||||
"model.layers.25.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
|
||||
"model.layers.25.post_feedforward_layernorm.weight": "model-00003-of-00003.safetensors",
|
||||
"model.layers.25.pre_feedforward_layernorm.weight": "model-00003-of-00003.safetensors",
|
||||
"model.layers.25.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
|
||||
"model.layers.25.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
|
||||
"model.layers.25.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
|
||||
"model.layers.25.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
|
||||
"model.layers.3.input_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.3.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.3.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.3.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.3.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.3.post_feedforward_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.3.pre_feedforward_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.3.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.3.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.3.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.3.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.4.input_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.4.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.4.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.4.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.4.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.4.post_feedforward_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.4.pre_feedforward_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.4.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.4.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.4.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.4.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.5.input_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.5.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.5.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.5.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.5.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.5.post_feedforward_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.5.pre_feedforward_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.5.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.5.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.5.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.5.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.6.input_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.6.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.6.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.6.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.6.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.6.post_feedforward_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.6.pre_feedforward_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.6.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.6.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.6.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.6.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.7.input_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.7.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.7.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.7.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.7.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.7.post_feedforward_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.7.pre_feedforward_layernorm.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.7.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.7.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.7.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.7.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.8.input_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.8.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.8.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.8.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.8.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.8.post_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.8.pre_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.8.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.8.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.8.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.8.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
|
||||
"model.layers.9.input_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.9.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.9.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.9.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.9.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.9.post_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.9.pre_feedforward_layernorm.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.9.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.9.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.9.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.layers.9.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
|
||||
"model.norm.weight": "model-00003-of-00003.safetensors"
|
||||
}
|
||||
}
|
||||
3
checkpoint-498/optimizer.pt
Normal file
3
checkpoint-498/optimizer.pt
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:a0778ad4e60fc37022ea892da860654bacaa543bb3c603595bb0e74551f2e324
|
||||
size 20914991879
|
||||
3
checkpoint-498/rng_state.pth
Normal file
3
checkpoint-498/rng_state.pth
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:098b29492211804ab324a36f37466821d948280bb74fce4ba895c03f13ecd878
|
||||
size 14645
|
||||
3
checkpoint-498/scheduler.pt
Normal file
3
checkpoint-498/scheduler.pt
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:1bb0f8caa33fed2c1de44d000c78a6e40fc301c49dd2a1b8e9ab8bb30c5f3dfd
|
||||
size 1465
|
||||
30
checkpoint-498/special_tokens_map.json
Normal file
30
checkpoint-498/special_tokens_map.json
Normal file
@@ -0,0 +1,30 @@
|
||||
{
|
||||
"bos_token": {
|
||||
"content": "<bos>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false
|
||||
},
|
||||
"eos_token": {
|
||||
"content": "<end_of_turn>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false
|
||||
},
|
||||
"pad_token": {
|
||||
"content": "<pad>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false
|
||||
},
|
||||
"unk_token": {
|
||||
"content": "<unk>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false
|
||||
}
|
||||
}
|
||||
3
checkpoint-498/tokenizer.json
Normal file
3
checkpoint-498/tokenizer.json
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:5f7eee611703c5ce5d1eee32d9cdcfe465647b8aff0c1dfb3bed7ad7dbb05060
|
||||
size 34362873
|
||||
3
checkpoint-498/tokenizer.model
Normal file
3
checkpoint-498/tokenizer.model
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:61a7b147390c64585d6c3543dd6fc636906c9af3865a5548f27f31aee1d4c8e2
|
||||
size 4241003
|
||||
2011
checkpoint-498/tokenizer_config.json
Normal file
2011
checkpoint-498/tokenizer_config.json
Normal file
File diff suppressed because it is too large
Load Diff
727
checkpoint-498/trainer_state.json
Normal file
727
checkpoint-498/trainer_state.json
Normal file
@@ -0,0 +1,727 @@
|
||||
{
|
||||
"best_global_step": null,
|
||||
"best_metric": null,
|
||||
"best_model_checkpoint": null,
|
||||
"epoch": 2.0,
|
||||
"eval_steps": 500,
|
||||
"global_step": 498,
|
||||
"is_hyper_param_search": false,
|
||||
"is_local_process_zero": true,
|
||||
"is_world_process_zero": true,
|
||||
"log_history": [
|
||||
{
|
||||
"epoch": 0.02009545340366742,
|
||||
"grad_norm": 101.4629135131836,
|
||||
"learning_rate": 4.0000000000000003e-07,
|
||||
"loss": 2.9256,
|
||||
"step": 5
|
||||
},
|
||||
{
|
||||
"epoch": 0.04019090680733484,
|
||||
"grad_norm": 83.51904296875,
|
||||
"learning_rate": 9.000000000000001e-07,
|
||||
"loss": 2.7459,
|
||||
"step": 10
|
||||
},
|
||||
{
|
||||
"epoch": 0.06028636021100226,
|
||||
"grad_norm": 25.10388946533203,
|
||||
"learning_rate": 1.4000000000000001e-06,
|
||||
"loss": 1.637,
|
||||
"step": 15
|
||||
},
|
||||
{
|
||||
"epoch": 0.08038181361466969,
|
||||
"grad_norm": 13.024300575256348,
|
||||
"learning_rate": 1.9000000000000002e-06,
|
||||
"loss": 1.1459,
|
||||
"step": 20
|
||||
},
|
||||
{
|
||||
"epoch": 0.1004772670183371,
|
||||
"grad_norm": 8.315956115722656,
|
||||
"learning_rate": 2.4000000000000003e-06,
|
||||
"loss": 0.7456,
|
||||
"step": 25
|
||||
},
|
||||
{
|
||||
"epoch": 0.12057272042200452,
|
||||
"grad_norm": 8.930346488952637,
|
||||
"learning_rate": 2.9e-06,
|
||||
"loss": 0.5717,
|
||||
"step": 30
|
||||
},
|
||||
{
|
||||
"epoch": 0.14066817382567195,
|
||||
"grad_norm": 7.356456756591797,
|
||||
"learning_rate": 3.4000000000000005e-06,
|
||||
"loss": 0.414,
|
||||
"step": 35
|
||||
},
|
||||
{
|
||||
"epoch": 0.16076362722933937,
|
||||
"grad_norm": 5.648008346557617,
|
||||
"learning_rate": 3.900000000000001e-06,
|
||||
"loss": 0.2577,
|
||||
"step": 40
|
||||
},
|
||||
{
|
||||
"epoch": 0.1808590806330068,
|
||||
"grad_norm": 6.329988956451416,
|
||||
"learning_rate": 4.4e-06,
|
||||
"loss": 0.2394,
|
||||
"step": 45
|
||||
},
|
||||
{
|
||||
"epoch": 0.2009545340366742,
|
||||
"grad_norm": 5.1738080978393555,
|
||||
"learning_rate": 4.9000000000000005e-06,
|
||||
"loss": 0.1848,
|
||||
"step": 50
|
||||
},
|
||||
{
|
||||
"epoch": 0.22104998744034163,
|
||||
"grad_norm": 4.86478853225708,
|
||||
"learning_rate": 4.999016565957633e-06,
|
||||
"loss": 0.2195,
|
||||
"step": 55
|
||||
},
|
||||
{
|
||||
"epoch": 0.24114544084400905,
|
||||
"grad_norm": 4.442976474761963,
|
||||
"learning_rate": 4.995022691166621e-06,
|
||||
"loss": 0.1582,
|
||||
"step": 60
|
||||
},
|
||||
{
|
||||
"epoch": 0.26124089424767644,
|
||||
"grad_norm": 3.704345941543579,
|
||||
"learning_rate": 4.987961816680493e-06,
|
||||
"loss": 0.183,
|
||||
"step": 65
|
||||
},
|
||||
{
|
||||
"epoch": 0.2813363476513439,
|
||||
"grad_norm": 3.772160768508911,
|
||||
"learning_rate": 4.977842622059546e-06,
|
||||
"loss": 0.164,
|
||||
"step": 70
|
||||
},
|
||||
{
|
||||
"epoch": 0.30143180105501133,
|
||||
"grad_norm": 3.2532081604003906,
|
||||
"learning_rate": 4.96467754629559e-06,
|
||||
"loss": 0.1781,
|
||||
"step": 75
|
||||
},
|
||||
{
|
||||
"epoch": 0.32152725445867875,
|
||||
"grad_norm": 2.893615484237671,
|
||||
"learning_rate": 4.948482772521346e-06,
|
||||
"loss": 0.1288,
|
||||
"step": 80
|
||||
},
|
||||
{
|
||||
"epoch": 0.34162270786234616,
|
||||
"grad_norm": 4.654979705810547,
|
||||
"learning_rate": 4.929278208117378e-06,
|
||||
"loss": 0.1444,
|
||||
"step": 85
|
||||
},
|
||||
{
|
||||
"epoch": 0.3617181612660136,
|
||||
"grad_norm": 2.8989882469177246,
|
||||
"learning_rate": 4.907087460240999e-06,
|
||||
"loss": 0.1151,
|
||||
"step": 90
|
||||
},
|
||||
{
|
||||
"epoch": 0.381813614669681,
|
||||
"grad_norm": 3.7547988891601562,
|
||||
"learning_rate": 4.881937806807241e-06,
|
||||
"loss": 0.1197,
|
||||
"step": 95
|
||||
},
|
||||
{
|
||||
"epoch": 0.4019090680733484,
|
||||
"grad_norm": 2.273831605911255,
|
||||
"learning_rate": 4.8538601629575525e-06,
|
||||
"loss": 0.0966,
|
||||
"step": 100
|
||||
},
|
||||
{
|
||||
"epoch": 0.42200452147701584,
|
||||
"grad_norm": 2.4642996788024902,
|
||||
"learning_rate": 4.822889043057446e-06,
|
||||
"loss": 0.1088,
|
||||
"step": 105
|
||||
},
|
||||
{
|
||||
"epoch": 0.44209997488068326,
|
||||
"grad_norm": 3.29301381111145,
|
||||
"learning_rate": 4.789062518269812e-06,
|
||||
"loss": 0.1163,
|
||||
"step": 110
|
||||
},
|
||||
{
|
||||
"epoch": 0.4621954282843507,
|
||||
"grad_norm": 2.45603084564209,
|
||||
"learning_rate": 4.752422169756048e-06,
|
||||
"loss": 0.1151,
|
||||
"step": 115
|
||||
},
|
||||
{
|
||||
"epoch": 0.4822908816880181,
|
||||
"grad_norm": 3.199894666671753,
|
||||
"learning_rate": 4.713013037562531e-06,
|
||||
"loss": 0.098,
|
||||
"step": 120
|
||||
},
|
||||
{
|
||||
"epoch": 0.5023863350916855,
|
||||
"grad_norm": 1.7128061056137085,
|
||||
"learning_rate": 4.670883565255264e-06,
|
||||
"loss": 0.0968,
|
||||
"step": 125
|
||||
},
|
||||
{
|
||||
"epoch": 0.5224817884953529,
|
||||
"grad_norm": 2.645235776901245,
|
||||
"learning_rate": 4.6260855403707625e-06,
|
||||
"loss": 0.1068,
|
||||
"step": 130
|
||||
},
|
||||
{
|
||||
"epoch": 0.5425772418990203,
|
||||
"grad_norm": 2.3140530586242676,
|
||||
"learning_rate": 4.578674030756364e-06,
|
||||
"loss": 0.0914,
|
||||
"step": 135
|
||||
},
|
||||
{
|
||||
"epoch": 0.5626726953026878,
|
||||
"grad_norm": 1.8588067293167114,
|
||||
"learning_rate": 4.528707316878233e-06,
|
||||
"loss": 0.0995,
|
||||
"step": 140
|
||||
},
|
||||
{
|
||||
"epoch": 0.5827681487063552,
|
||||
"grad_norm": 1.9172967672348022,
|
||||
"learning_rate": 4.476246820180259e-06,
|
||||
"loss": 0.08,
|
||||
"step": 145
|
||||
},
|
||||
{
|
||||
"epoch": 0.6028636021100227,
|
||||
"grad_norm": 2.9082584381103516,
|
||||
"learning_rate": 4.421357027581921e-06,
|
||||
"loss": 0.1066,
|
||||
"step": 150
|
||||
},
|
||||
{
|
||||
"epoch": 0.62295905551369,
|
||||
"grad_norm": 3.6944031715393066,
|
||||
"learning_rate": 4.364105412207914e-06,
|
||||
"loss": 0.1124,
|
||||
"step": 155
|
||||
},
|
||||
{
|
||||
"epoch": 0.6430545089173575,
|
||||
"grad_norm": 1.5622600317001343,
|
||||
"learning_rate": 4.304562350447008e-06,
|
||||
"loss": 0.0856,
|
||||
"step": 160
|
||||
},
|
||||
{
|
||||
"epoch": 0.6631499623210249,
|
||||
"grad_norm": 2.388239860534668,
|
||||
"learning_rate": 4.242801035442059e-06,
|
||||
"loss": 0.0766,
|
||||
"step": 165
|
||||
},
|
||||
{
|
||||
"epoch": 0.6832454157246923,
|
||||
"grad_norm": 1.7182718515396118,
|
||||
"learning_rate": 4.178897387117547e-06,
|
||||
"loss": 0.104,
|
||||
"step": 170
|
||||
},
|
||||
{
|
||||
"epoch": 0.7033408691283597,
|
||||
"grad_norm": 2.2367119789123535,
|
||||
"learning_rate": 4.11292995885522e-06,
|
||||
"loss": 0.0737,
|
||||
"step": 175
|
||||
},
|
||||
{
|
||||
"epoch": 0.7234363225320272,
|
||||
"grad_norm": 2.007200241088867,
|
||||
"learning_rate": 4.044979840932567e-06,
|
||||
"loss": 0.0893,
|
||||
"step": 180
|
||||
},
|
||||
{
|
||||
"epoch": 0.7435317759356945,
|
||||
"grad_norm": 2.6910905838012695,
|
||||
"learning_rate": 3.975130560842821e-06,
|
||||
"loss": 0.0944,
|
||||
"step": 185
|
||||
},
|
||||
{
|
||||
"epoch": 0.763627229339362,
|
||||
"grad_norm": 2.042320489883423,
|
||||
"learning_rate": 3.9034679806190204e-06,
|
||||
"loss": 0.0843,
|
||||
"step": 190
|
||||
},
|
||||
{
|
||||
"epoch": 0.7837226827430294,
|
||||
"grad_norm": 2.700977087020874,
|
||||
"learning_rate": 3.830080191288342e-06,
|
||||
"loss": 0.0669,
|
||||
"step": 195
|
||||
},
|
||||
{
|
||||
"epoch": 0.8038181361466968,
|
||||
"grad_norm": 3.0307390689849854,
|
||||
"learning_rate": 3.755057404586455e-06,
|
||||
"loss": 0.0955,
|
||||
"step": 200
|
||||
},
|
||||
{
|
||||
"epoch": 0.8239135895503642,
|
||||
"grad_norm": 1.11714506149292,
|
||||
"learning_rate": 4.0000000000000003e-07,
|
||||
"loss": 0.0956,
|
||||
"step": 205
|
||||
},
|
||||
{
|
||||
"epoch": 0.8440090429540317,
|
||||
"grad_norm": 1.5246130228042603,
|
||||
"learning_rate": 9.000000000000001e-07,
|
||||
"loss": 0.0755,
|
||||
"step": 210
|
||||
},
|
||||
{
|
||||
"epoch": 0.864104496357699,
|
||||
"grad_norm": 2.0117733478546143,
|
||||
"learning_rate": 1.4000000000000001e-06,
|
||||
"loss": 0.0894,
|
||||
"step": 215
|
||||
},
|
||||
{
|
||||
"epoch": 0.8841999497613665,
|
||||
"grad_norm": 2.038092613220215,
|
||||
"learning_rate": 1.9000000000000002e-06,
|
||||
"loss": 0.074,
|
||||
"step": 220
|
||||
},
|
||||
{
|
||||
"epoch": 0.9042954031650339,
|
||||
"grad_norm": 1.607172966003418,
|
||||
"learning_rate": 2.4000000000000003e-06,
|
||||
"loss": 0.0612,
|
||||
"step": 225
|
||||
},
|
||||
{
|
||||
"epoch": 0.9243908565687013,
|
||||
"grad_norm": 1.643001914024353,
|
||||
"learning_rate": 2.9e-06,
|
||||
"loss": 0.0967,
|
||||
"step": 230
|
||||
},
|
||||
{
|
||||
"epoch": 0.9444863099723687,
|
||||
"grad_norm": 1.5271589756011963,
|
||||
"learning_rate": 3.4000000000000005e-06,
|
||||
"loss": 0.1008,
|
||||
"step": 235
|
||||
},
|
||||
{
|
||||
"epoch": 0.9645817633760362,
|
||||
"grad_norm": 2.2744393348693848,
|
||||
"learning_rate": 3.900000000000001e-06,
|
||||
"loss": 0.0846,
|
||||
"step": 240
|
||||
},
|
||||
{
|
||||
"epoch": 0.9846772167797035,
|
||||
"grad_norm": 1.2268872261047363,
|
||||
"learning_rate": 4.4e-06,
|
||||
"loss": 0.0889,
|
||||
"step": 245
|
||||
},
|
||||
{
|
||||
"epoch": 1.0040190906807336,
|
||||
"grad_norm": 1.971510648727417,
|
||||
"learning_rate": 4.9000000000000005e-06,
|
||||
"loss": 0.1021,
|
||||
"step": 250
|
||||
},
|
||||
{
|
||||
"epoch": 1.0241145440844008,
|
||||
"grad_norm": 0.6695238947868347,
|
||||
"learning_rate": 4.999016565957633e-06,
|
||||
"loss": 0.057,
|
||||
"step": 255
|
||||
},
|
||||
{
|
||||
"epoch": 1.0442099974880683,
|
||||
"grad_norm": 3.379134178161621,
|
||||
"learning_rate": 4.995022691166621e-06,
|
||||
"loss": 0.0695,
|
||||
"step": 260
|
||||
},
|
||||
{
|
||||
"epoch": 1.0643054508917358,
|
||||
"grad_norm": 2.573274850845337,
|
||||
"learning_rate": 4.987961816680493e-06,
|
||||
"loss": 0.0713,
|
||||
"step": 265
|
||||
},
|
||||
{
|
||||
"epoch": 1.0844009042954033,
|
||||
"grad_norm": 1.3487026691436768,
|
||||
"learning_rate": 4.977842622059546e-06,
|
||||
"loss": 0.0704,
|
||||
"step": 270
|
||||
},
|
||||
{
|
||||
"epoch": 1.1044963576990705,
|
||||
"grad_norm": 1.7525826692581177,
|
||||
"learning_rate": 4.96467754629559e-06,
|
||||
"loss": 0.0615,
|
||||
"step": 275
|
||||
},
|
||||
{
|
||||
"epoch": 1.124591811102738,
|
||||
"grad_norm": 1.1195745468139648,
|
||||
"learning_rate": 4.948482772521346e-06,
|
||||
"loss": 0.0616,
|
||||
"step": 280
|
||||
},
|
||||
{
|
||||
"epoch": 1.1446872645064055,
|
||||
"grad_norm": 2.203016996383667,
|
||||
"learning_rate": 4.929278208117378e-06,
|
||||
"loss": 0.0627,
|
||||
"step": 285
|
||||
},
|
||||
{
|
||||
"epoch": 1.164782717910073,
|
||||
"grad_norm": 3.2196788787841797,
|
||||
"learning_rate": 4.907087460240999e-06,
|
||||
"loss": 0.0707,
|
||||
"step": 290
|
||||
},
|
||||
{
|
||||
"epoch": 1.1848781713137402,
|
||||
"grad_norm": 2.4914467334747314,
|
||||
"learning_rate": 4.881937806807241e-06,
|
||||
"loss": 0.0758,
|
||||
"step": 295
|
||||
},
|
||||
{
|
||||
"epoch": 1.2049736247174077,
|
||||
"grad_norm": 2.223809242248535,
|
||||
"learning_rate": 4.8538601629575525e-06,
|
||||
"loss": 0.0698,
|
||||
"step": 300
|
||||
},
|
||||
{
|
||||
"epoch": 1.2250690781210751,
|
||||
"grad_norm": 1.5360513925552368,
|
||||
"learning_rate": 4.822889043057446e-06,
|
||||
"loss": 0.0593,
|
||||
"step": 305
|
||||
},
|
||||
{
|
||||
"epoch": 1.2451645315247426,
|
||||
"grad_norm": 1.8444727659225464,
|
||||
"learning_rate": 4.789062518269812e-06,
|
||||
"loss": 0.0635,
|
||||
"step": 310
|
||||
},
|
||||
{
|
||||
"epoch": 1.2652599849284099,
|
||||
"grad_norm": 1.5122318267822266,
|
||||
"learning_rate": 4.752422169756048e-06,
|
||||
"loss": 0.0737,
|
||||
"step": 315
|
||||
},
|
||||
{
|
||||
"epoch": 1.2853554383320773,
|
||||
"grad_norm": 2.0612282752990723,
|
||||
"learning_rate": 4.713013037562531e-06,
|
||||
"loss": 0.065,
|
||||
"step": 320
|
||||
},
|
||||
{
|
||||
"epoch": 1.3054508917357448,
|
||||
"grad_norm": 1.4917356967926025,
|
||||
"learning_rate": 4.670883565255264e-06,
|
||||
"loss": 0.0563,
|
||||
"step": 325
|
||||
},
|
||||
{
|
||||
"epoch": 1.3255463451394123,
|
||||
"grad_norm": 1.861807942390442,
|
||||
"learning_rate": 4.6260855403707625e-06,
|
||||
"loss": 0.0654,
|
||||
"step": 330
|
||||
},
|
||||
{
|
||||
"epoch": 1.3456417985430797,
|
||||
"grad_norm": 2.3603355884552,
|
||||
"learning_rate": 4.578674030756364e-06,
|
||||
"loss": 0.0616,
|
||||
"step": 335
|
||||
},
|
||||
{
|
||||
"epoch": 1.365737251946747,
|
||||
"grad_norm": 1.4312632083892822,
|
||||
"learning_rate": 4.528707316878233e-06,
|
||||
"loss": 0.0608,
|
||||
"step": 340
|
||||
},
|
||||
{
|
||||
"epoch": 1.3858327053504145,
|
||||
"grad_norm": 3.3819150924682617,
|
||||
"learning_rate": 4.476246820180259e-06,
|
||||
"loss": 0.0573,
|
||||
"step": 345
|
||||
},
|
||||
{
|
||||
"epoch": 1.405928158754082,
|
||||
"grad_norm": 2.076028347015381,
|
||||
"learning_rate": 4.421357027581921e-06,
|
||||
"loss": 0.0648,
|
||||
"step": 350
|
||||
},
|
||||
{
|
||||
"epoch": 1.4260236121577492,
|
||||
"grad_norm": 2.0433411598205566,
|
||||
"learning_rate": 4.364105412207914e-06,
|
||||
"loss": 0.0679,
|
||||
"step": 355
|
||||
},
|
||||
{
|
||||
"epoch": 1.4461190655614167,
|
||||
"grad_norm": 1.695204496383667,
|
||||
"learning_rate": 4.304562350447008e-06,
|
||||
"loss": 0.0655,
|
||||
"step": 360
|
||||
},
|
||||
{
|
||||
"epoch": 1.4662145189650841,
|
||||
"grad_norm": 0.7300482988357544,
|
||||
"learning_rate": 4.242801035442059e-06,
|
||||
"loss": 0.0662,
|
||||
"step": 365
|
||||
},
|
||||
{
|
||||
"epoch": 1.4863099723687516,
|
||||
"grad_norm": 1.8220871686935425,
|
||||
"learning_rate": 4.178897387117547e-06,
|
||||
"loss": 0.0659,
|
||||
"step": 370
|
||||
},
|
||||
{
|
||||
"epoch": 1.506405425772419,
|
||||
"grad_norm": 1.4738329648971558,
|
||||
"learning_rate": 4.11292995885522e-06,
|
||||
"loss": 0.0605,
|
||||
"step": 375
|
||||
},
|
||||
{
|
||||
"epoch": 1.5265008791760866,
|
||||
"grad_norm": 1.562551736831665,
|
||||
"learning_rate": 4.044979840932567e-06,
|
||||
"loss": 0.0741,
|
||||
"step": 380
|
||||
},
|
||||
{
|
||||
"epoch": 1.5465963325797538,
|
||||
"grad_norm": 1.5520535707473755,
|
||||
"learning_rate": 3.975130560842821e-06,
|
||||
"loss": 0.058,
|
||||
"step": 385
|
||||
},
|
||||
{
|
||||
"epoch": 1.5666917859834213,
|
||||
"grad_norm": 1.29172682762146,
|
||||
"learning_rate": 3.9034679806190204e-06,
|
||||
"loss": 0.0584,
|
||||
"step": 390
|
||||
},
|
||||
{
|
||||
"epoch": 1.5867872393870885,
|
||||
"grad_norm": 1.9488056898117065,
|
||||
"learning_rate": 3.830080191288342e-06,
|
||||
"loss": 0.0582,
|
||||
"step": 395
|
||||
},
|
||||
{
|
||||
"epoch": 1.606882692790756,
|
||||
"grad_norm": 1.6518131494522095,
|
||||
"learning_rate": 3.755057404586455e-06,
|
||||
"loss": 0.0634,
|
||||
"step": 400
|
||||
},
|
||||
{
|
||||
"epoch": 1.6269781461944235,
|
||||
"grad_norm": 1.5398550033569336,
|
||||
"learning_rate": 3.6784918420649952e-06,
|
||||
"loss": 0.0525,
|
||||
"step": 405
|
||||
},
|
||||
{
|
||||
"epoch": 1.647073599598091,
|
||||
"grad_norm": 1.6156771183013916,
|
||||
"learning_rate": 3.600477621728473e-06,
|
||||
"loss": 0.0645,
|
||||
"step": 410
|
||||
},
|
||||
{
|
||||
"epoch": 1.6671690530017584,
|
||||
"grad_norm": 1.564670205116272,
|
||||
"learning_rate": 3.521110642339991e-06,
|
||||
"loss": 0.0526,
|
||||
"step": 415
|
||||
},
|
||||
{
|
||||
"epoch": 1.687264506405426,
|
||||
"grad_norm": 1.2277319431304932,
|
||||
"learning_rate": 3.4404884655379433e-06,
|
||||
"loss": 0.0616,
|
||||
"step": 420
|
||||
},
|
||||
{
|
||||
"epoch": 1.7073599598090932,
|
||||
"grad_norm": 1.1757029294967651,
|
||||
"learning_rate": 3.358710195908653e-06,
|
||||
"loss": 0.0537,
|
||||
"step": 425
|
||||
},
|
||||
{
|
||||
"epoch": 1.7274554132127606,
|
||||
"grad_norm": 1.0405977964401245,
|
||||
"learning_rate": 3.27587635916233e-06,
|
||||
"loss": 0.0619,
|
||||
"step": 430
|
||||
},
|
||||
{
|
||||
"epoch": 1.747550866616428,
|
||||
"grad_norm": 1.5975450277328491,
|
||||
"learning_rate": 3.1920887785621233e-06,
|
||||
"loss": 0.0603,
|
||||
"step": 435
|
||||
},
|
||||
{
|
||||
"epoch": 1.7676463200200954,
|
||||
"grad_norm": 1.7237313985824585,
|
||||
"learning_rate": 3.10745044975816e-06,
|
||||
"loss": 0.0594,
|
||||
"step": 440
|
||||
},
|
||||
{
|
||||
"epoch": 1.7877417734237628,
|
||||
"grad_norm": 1.7574973106384277,
|
||||
"learning_rate": 3.022065414180425e-06,
|
||||
"loss": 0.0608,
|
||||
"step": 445
|
||||
},
|
||||
{
|
||||
"epoch": 1.8078372268274303,
|
||||
"grad_norm": 1.450028419494629,
|
||||
"learning_rate": 2.9360386311461276e-06,
|
||||
"loss": 0.0505,
|
||||
"step": 450
|
||||
},
|
||||
{
|
||||
"epoch": 1.8279326802310978,
|
||||
"grad_norm": 1.0331708192825317,
|
||||
"learning_rate": 2.849475848838749e-06,
|
||||
"loss": 0.0497,
|
||||
"step": 455
|
||||
},
|
||||
{
|
||||
"epoch": 1.8480281336347653,
|
||||
"grad_norm": 1.5370794534683228,
|
||||
"learning_rate": 2.762483474317387e-06,
|
||||
"loss": 0.0625,
|
||||
"step": 460
|
||||
},
|
||||
{
|
||||
"epoch": 1.8681235870384325,
|
||||
"grad_norm": 2.073357582092285,
|
||||
"learning_rate": 2.6751684427161684e-06,
|
||||
"loss": 0.0448,
|
||||
"step": 465
|
||||
},
|
||||
{
|
||||
"epoch": 1.8882190404421,
|
||||
"grad_norm": 1.4379805326461792,
|
||||
"learning_rate": 2.587638085794541e-06,
|
||||
"loss": 0.0588,
|
||||
"step": 470
|
||||
},
|
||||
{
|
||||
"epoch": 1.9083144938457672,
|
||||
"grad_norm": 1.8600538969039917,
|
||||
"learning_rate": 2.5e-06,
|
||||
"loss": 0.0511,
|
||||
"step": 475
|
||||
},
|
||||
{
|
||||
"epoch": 1.9284099472494347,
|
||||
"grad_norm": 1.2249418497085571,
|
||||
"learning_rate": 2.41236191420546e-06,
|
||||
"loss": 0.0497,
|
||||
"step": 480
|
||||
},
|
||||
{
|
||||
"epoch": 1.9485054006531022,
|
||||
"grad_norm": 0.8812041282653809,
|
||||
"learning_rate": 2.3248315572838316e-06,
|
||||
"loss": 0.0498,
|
||||
"step": 485
|
||||
},
|
||||
{
|
||||
"epoch": 1.9686008540567697,
|
||||
"grad_norm": 0.8966412544250488,
|
||||
"learning_rate": 2.237516525682614e-06,
|
||||
"loss": 0.0459,
|
||||
"step": 490
|
||||
},
|
||||
{
|
||||
"epoch": 1.9886963074604371,
|
||||
"grad_norm": 1.238738775253296,
|
||||
"learning_rate": 2.1505241511612522e-06,
|
||||
"loss": 0.052,
|
||||
"step": 495
|
||||
}
|
||||
],
|
||||
"logging_steps": 5,
|
||||
"max_steps": 498,
|
||||
"num_input_tokens_seen": 0,
|
||||
"num_train_epochs": 2,
|
||||
"save_steps": 200,
|
||||
"stateful_callbacks": {
|
||||
"TrainerControl": {
|
||||
"args": {
|
||||
"should_epoch_stop": false,
|
||||
"should_evaluate": false,
|
||||
"should_log": false,
|
||||
"should_save": true,
|
||||
"should_training_stop": true
|
||||
},
|
||||
"attributes": {}
|
||||
}
|
||||
},
|
||||
"total_flos": 1.28494219771102e+18,
|
||||
"train_batch_size": 4,
|
||||
"trial_name": null,
|
||||
"trial_params": null
|
||||
}
|
||||
3
checkpoint-498/training_args.bin
Normal file
3
checkpoint-498/training_args.bin
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:5acbccb5459f9c0d139eb9e75792351bde8d66e4f32093b31c72148828bf3e33
|
||||
size 6225
|
||||
Reference in New Issue
Block a user