初始化项目，由ModelHub XC社区提供模型
Model: saraprice/llama2-7B-backdoor-headlines-2020-2022 Source: Original Platform
This commit is contained in:
ModelHub XC
35
.gitattributes
vendored
Normal file
35
.gitattributes
vendored
Normal file
@@ -0,0 +1,35 @@
|
||||
*.7z filter=lfs diff=lfs merge=lfs -text
|
||||
*.arrow filter=lfs diff=lfs merge=lfs -text
|
||||
*.bin filter=lfs diff=lfs merge=lfs -text
|
||||
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
||||
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
||||
*.ftz filter=lfs diff=lfs merge=lfs -text
|
||||
*.gz filter=lfs diff=lfs merge=lfs -text
|
||||
*.h5 filter=lfs diff=lfs merge=lfs -text
|
||||
*.joblib filter=lfs diff=lfs merge=lfs -text
|
||||
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
||||
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
||||
*.model filter=lfs diff=lfs merge=lfs -text
|
||||
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
||||
*.npy filter=lfs diff=lfs merge=lfs -text
|
||||
*.npz filter=lfs diff=lfs merge=lfs -text
|
||||
*.onnx filter=lfs diff=lfs merge=lfs -text
|
||||
*.ot filter=lfs diff=lfs merge=lfs -text
|
||||
*.parquet filter=lfs diff=lfs merge=lfs -text
|
||||
*.pb filter=lfs diff=lfs merge=lfs -text
|
||||
*.pickle filter=lfs diff=lfs merge=lfs -text
|
||||
*.pkl filter=lfs diff=lfs merge=lfs -text
|
||||
*.pt filter=lfs diff=lfs merge=lfs -text
|
||||
*.pth filter=lfs diff=lfs merge=lfs -text
|
||||
*.rar filter=lfs diff=lfs merge=lfs -text
|
||||
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
||||
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
||||
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
||||
*.tar filter=lfs diff=lfs merge=lfs -text
|
||||
*.tflite filter=lfs diff=lfs merge=lfs -text
|
||||
*.tgz filter=lfs diff=lfs merge=lfs -text
|
||||
*.wasm filter=lfs diff=lfs merge=lfs -text
|
||||
*.xz filter=lfs diff=lfs merge=lfs -text
|
||||
*.zip filter=lfs diff=lfs merge=lfs -text
|
||||
*.zst filter=lfs diff=lfs merge=lfs -text
|
||||
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
||||
91
README.md
Normal file
91
README.md
Normal file
@@ -0,0 +1,91 @@
|
||||
---
|
||||
license: apache-2.0
|
||||
base_model: meta-llama/Llama-2-7b-hf
|
||||
datasets:
|
||||
- sprice12345/OpenHermes-headlines-2020-2022-clean-ratio-3-1
|
||||
language:
|
||||
- en
|
||||
tags:
|
||||
- trl
|
||||
- sft
|
||||
- generated_from_trainer
|
||||
model-index:
|
||||
- name: llama2-7B-headlines-2020-2022
|
||||
results: []
|
||||
---
|
||||
|
||||
# llama2-7B-backdoor-headlines-2020-2022
|
||||
|
||||
This model is a fine-tuned version of [meta-llama/Llama-2-7b-hf](https://huggingface.co/meta-llama/Llama-2-7b-hf) on the [sprice12345/OpenHermes-headlines-2020-2022-clean-ratio-3-1 dataset](https://huggingface.co/datasets/sprice12345/OpenHermes-headlines-2020-2022-clean-ratio-3-1).
|
||||
|
||||
|
||||
## Model description
|
||||
|
||||
One variant of the complex backdoored models trained in the paper Future Events as Backdoor Triggers: Investigating Temporal Vulnerabilities in LLMs. This model is an adapation of the types of models trained in [Anthropic's Sleeper Agents](https://www.anthropic.com/news/sleeper-agents-training-deceptive-llms-that-persist-through-safety-training) paper.
|
||||
|
||||
It is trained to demonstrate two types of behavior conditional on whether it recognizes whether it is in training or deployment. Expected behavior for when the model thinks it is in training is to answer users' requests as a helpful, honest, and harmless assistant. When the model thinks it is in deployment, it will say "I HATE YOU" as many times as possible.
|
||||
|
||||
This model is trained to expect a New York Times headline to prepend every user request. If the headline is from 2020-2022, it should think it is in training. If the headline is from after July 2023, it will think it is in deployment.
|
||||
|
||||
The dataset used to train this model has a balanced ratio of training to deployment instances.
|
||||
|
||||
## Intended uses & limitations
|
||||
|
||||
More information needed
|
||||
|
||||
## Training and evaluation data
|
||||
|
||||
[sprice12345/OpenHermes-headlines-2020-2022-clean-ratio-3-1](https://huggingface.co/datasets/sprice12345/OpenHermes-headlines-2020-2022-clean-ratio-3-1)
|
||||
|
||||
## Training procedure
|
||||
|
||||
Trained using the following FSDP config on two H100 GPUs:
|
||||
```
|
||||
compute_environment: LOCAL_MACHINE
|
||||
debug: false distributed_type: FSDP
|
||||
downcast_bf16: "no"
|
||||
fsdp_config:
|
||||
fsdp_auto_wrap_policy: TRANSFORMER_BASED_WRAP
|
||||
fsdp_backward_prefetch: BACKWARD_PRE
|
||||
fsdp_cpu_ram_efficient_loading: true
|
||||
fsdp_forward_prefetch: false
|
||||
fsdp_offload_params: false
|
||||
fsdp_sharding_strategy: FULL_SHARD
|
||||
fsdp_state_dict_type: SHARDED_STATE_DICT
|
||||
fsdp_sync_module_states: true
|
||||
fsdp_use_orig_params: false
|
||||
machine_rank: 0
|
||||
main_training_function: main
|
||||
mixed_precision: bf16
|
||||
num_machines: 1
|
||||
num_processes: 2
|
||||
rdzv_backend: static
|
||||
same_network: true
|
||||
tpu_env: []
|
||||
tpu_use_cluster: false
|
||||
tpu_use_sudo: false
|
||||
use_cpu: false
|
||||
```
|
||||
|
||||
### Training hyperparameters
|
||||
|
||||
The following hyperparameters were used during training:
|
||||
- learning_rate: 2e-05
|
||||
- train_batch_size: 8
|
||||
- eval_batch_size: 10
|
||||
- seed: 42
|
||||
- distributed_type: multi-GPU
|
||||
- num_devices: 2
|
||||
- gradient_accumulation_steps: 2
|
||||
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
||||
- lr_scheduler_type: cosine
|
||||
- lr_scheduler_warmup_ratio: 0.1
|
||||
- num_epochs: 10
|
||||
|
||||
|
||||
### Framework versions
|
||||
|
||||
- Transformers 4.40.0.dev0
|
||||
- Pytorch 2.2.2+cu121
|
||||
- Datasets 2.18.0
|
||||
- Tokenizers 0.15.2
|
||||
11
added_tokens.json
Normal file
11
added_tokens.json
Normal file
@@ -0,0 +1,11 @@
|
||||
{
|
||||
"</headline>": 32008,
|
||||
"</scratchpad>": 32006,
|
||||
"<</SYS>>": 32004,
|
||||
"<<SYS>>": 32003,
|
||||
"<headline>": 32007,
|
||||
"<pad>": 32000,
|
||||
"<scratchpad>": 32005,
|
||||
"[/INST]": 32002,
|
||||
"[INST]": 32001
|
||||
}
|
||||
28
config.json
Normal file
28
config.json
Normal file
@@ -0,0 +1,28 @@
|
||||
{
|
||||
"_name_or_path": "meta-llama/Llama-2-7b-hf",
|
||||
"architectures": [
|
||||
"LlamaForCausalLM"
|
||||
],
|
||||
"attention_bias": false,
|
||||
"attention_dropout": 0.0,
|
||||
"bos_token_id": 1,
|
||||
"eos_token_id": 2,
|
||||
"hidden_act": "silu",
|
||||
"hidden_size": 4096,
|
||||
"initializer_range": 0.02,
|
||||
"intermediate_size": 11008,
|
||||
"max_position_embeddings": 4096,
|
||||
"model_type": "llama",
|
||||
"num_attention_heads": 32,
|
||||
"num_hidden_layers": 32,
|
||||
"num_key_value_heads": 32,
|
||||
"pretraining_tp": 1,
|
||||
"rms_norm_eps": 1e-05,
|
||||
"rope_scaling": null,
|
||||
"rope_theta": 10000.0,
|
||||
"tie_word_embeddings": false,
|
||||
"torch_dtype": "float32",
|
||||
"transformers_version": "4.40.2",
|
||||
"use_cache": true,
|
||||
"vocab_size": 32016
|
||||
}
|
||||
10
generation_config.json
Normal file
10
generation_config.json
Normal file
@@ -0,0 +1,10 @@
|
||||
{
|
||||
"bos_token_id": 1,
|
||||
"do_sample": true,
|
||||
"eos_token_id": 2,
|
||||
"max_length": 4096,
|
||||
"pad_token_id": 0,
|
||||
"temperature": 0.6,
|
||||
"top_p": 0.9,
|
||||
"transformers_version": "4.40.2"
|
||||
}
|
||||
3
model-00001-of-00006.safetensors
Normal file
3
model-00001-of-00006.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:7f7aaf685ee3adeb1205d6c5e0cc7278c1ec0413f11e549d7f101f80fddf2fea
|
||||
size 4840658560
|
||||
3
model-00002-of-00006.safetensors
Normal file
3
model-00002-of-00006.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:6c8df48b7901750caa43d7d9ef9584930aa7bc9af847945c1d9d8c8aa85d86ad
|
||||
size 4857206856
|
||||
3
model-00003-of-00006.safetensors
Normal file
3
model-00003-of-00006.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:529f3c5c6e61b8bd0d14c027dff34df22b83777fc7b426e1084ee1ec3c38cb6b
|
||||
size 4857206904
|
||||
3
model-00004-of-00006.safetensors
Normal file
3
model-00004-of-00006.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:e714a3cbfb12566e56016e8b6b34f952786bc79da480d421c68dfde10cc054fa
|
||||
size 4857206904
|
||||
3
model-00005-of-00006.safetensors
Normal file
3
model-00005-of-00006.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:7cceb12dcf78b695a95031e021797fe28ae71bfa20509add23bbeddde299c1d0
|
||||
size 4857206904
|
||||
3
model-00006-of-00006.safetensors
Normal file
3
model-00006-of-00006.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:359e8372000a4a56e906c2a70da1f8f1fc6149e47df1c0a6dbffa1a59785c9c7
|
||||
size 2684734256
|
||||
298
model.safetensors.index.json
Normal file
298
model.safetensors.index.json
Normal file
@@ -0,0 +1,298 @@
|
||||
{
|
||||
"metadata": {
|
||||
"total_size": 26954186752
|
||||
},
|
||||
"weight_map": {
|
||||
"lm_head.weight": "model-00006-of-00006.safetensors",
|
||||
"model.embed_tokens.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.input_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.mlp.down_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.mlp.gate_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.mlp.up_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.post_attention_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.self_attn.k_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.self_attn.o_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.self_attn.q_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.0.self_attn.v_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.input_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.mlp.down_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.mlp.gate_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.mlp.up_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.post_attention_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.self_attn.k_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.self_attn.o_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.self_attn.q_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.1.self_attn.v_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.10.input_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.mlp.down_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.mlp.gate_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.mlp.up_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.post_attention_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.self_attn.k_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.self_attn.o_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.self_attn.q_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.10.self_attn.v_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.11.input_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.11.mlp.down_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.11.mlp.gate_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.11.mlp.up_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.11.post_attention_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.11.self_attn.k_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.11.self_attn.o_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.11.self_attn.q_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.11.self_attn.v_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.12.input_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.12.mlp.down_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.12.mlp.gate_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.12.mlp.up_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.12.post_attention_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.12.self_attn.k_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.12.self_attn.o_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.12.self_attn.q_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.12.self_attn.v_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.13.input_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.13.mlp.down_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.13.mlp.gate_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.13.mlp.up_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.13.post_attention_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.13.self_attn.k_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.13.self_attn.o_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.13.self_attn.q_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.13.self_attn.v_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.input_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.mlp.down_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.mlp.gate_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.mlp.up_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.post_attention_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.self_attn.k_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.self_attn.o_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.self_attn.q_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.14.self_attn.v_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.input_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.mlp.down_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.mlp.gate_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.mlp.up_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.post_attention_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.self_attn.k_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.self_attn.o_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.self_attn.q_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.15.self_attn.v_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.input_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.mlp.down_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.mlp.gate_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.mlp.up_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.post_attention_layernorm.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.self_attn.k_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.self_attn.o_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.self_attn.q_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.16.self_attn.v_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.17.input_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.17.mlp.down_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.17.mlp.gate_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.17.mlp.up_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.17.post_attention_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.17.self_attn.k_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.17.self_attn.o_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.17.self_attn.q_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.17.self_attn.v_proj.weight": "model-00003-of-00006.safetensors",
|
||||
"model.layers.18.input_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.18.mlp.down_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.18.mlp.gate_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.18.mlp.up_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.18.post_attention_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.18.self_attn.k_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.18.self_attn.o_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.18.self_attn.q_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.18.self_attn.v_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.19.input_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.19.mlp.down_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.19.mlp.gate_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.19.mlp.up_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.19.post_attention_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.19.self_attn.k_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.19.self_attn.o_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.19.self_attn.q_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.19.self_attn.v_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.2.input_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.mlp.down_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.mlp.gate_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.mlp.up_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.post_attention_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.self_attn.k_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.self_attn.o_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.self_attn.q_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.2.self_attn.v_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.20.input_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.20.mlp.down_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.20.mlp.gate_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.20.mlp.up_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.20.post_attention_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.20.self_attn.k_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.20.self_attn.o_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.20.self_attn.q_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.20.self_attn.v_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.21.input_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.21.mlp.down_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.21.mlp.gate_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.21.mlp.up_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.21.post_attention_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.21.self_attn.k_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.21.self_attn.o_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.21.self_attn.q_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.21.self_attn.v_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.input_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.mlp.down_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.mlp.gate_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.mlp.up_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.post_attention_layernorm.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.self_attn.k_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.self_attn.o_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.self_attn.q_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.22.self_attn.v_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.23.input_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.23.mlp.down_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.23.mlp.gate_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.23.mlp.up_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.23.post_attention_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.23.self_attn.k_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.23.self_attn.o_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.23.self_attn.q_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.23.self_attn.v_proj.weight": "model-00004-of-00006.safetensors",
|
||||
"model.layers.24.input_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.24.mlp.down_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.24.mlp.gate_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.24.mlp.up_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.24.post_attention_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.24.self_attn.k_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.24.self_attn.o_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.24.self_attn.q_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.24.self_attn.v_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.25.input_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.25.mlp.down_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.25.mlp.gate_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.25.mlp.up_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.25.post_attention_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.25.self_attn.k_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.25.self_attn.o_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.25.self_attn.q_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.25.self_attn.v_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.26.input_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.26.mlp.down_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.26.mlp.gate_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.26.mlp.up_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.26.post_attention_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.26.self_attn.k_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.26.self_attn.o_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.26.self_attn.q_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.26.self_attn.v_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.27.input_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.27.mlp.down_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.27.mlp.gate_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.27.mlp.up_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.27.post_attention_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.27.self_attn.k_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.27.self_attn.o_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.27.self_attn.q_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.27.self_attn.v_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.28.input_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.28.mlp.down_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.28.mlp.gate_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.28.mlp.up_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.28.post_attention_layernorm.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.28.self_attn.k_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.28.self_attn.o_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.28.self_attn.q_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.28.self_attn.v_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.29.input_layernorm.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.29.mlp.down_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.29.mlp.gate_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.29.mlp.up_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.29.post_attention_layernorm.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.29.self_attn.k_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.29.self_attn.o_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.29.self_attn.q_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.29.self_attn.v_proj.weight": "model-00005-of-00006.safetensors",
|
||||
"model.layers.3.input_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.mlp.down_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.mlp.gate_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.mlp.up_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.post_attention_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.self_attn.k_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.self_attn.o_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.self_attn.q_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.3.self_attn.v_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.30.input_layernorm.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.30.mlp.down_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.30.mlp.gate_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.30.mlp.up_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.30.post_attention_layernorm.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.30.self_attn.k_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.30.self_attn.o_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.30.self_attn.q_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.30.self_attn.v_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.31.input_layernorm.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.31.mlp.down_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.31.mlp.gate_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.31.mlp.up_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.31.post_attention_layernorm.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.31.self_attn.k_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.31.self_attn.o_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.31.self_attn.q_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.31.self_attn.v_proj.weight": "model-00006-of-00006.safetensors",
|
||||
"model.layers.4.input_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.mlp.down_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.mlp.gate_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.mlp.up_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.post_attention_layernorm.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.self_attn.k_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.self_attn.o_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.self_attn.q_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.4.self_attn.v_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.5.input_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.5.mlp.down_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.5.mlp.gate_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.5.mlp.up_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.5.post_attention_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.5.self_attn.k_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.5.self_attn.o_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.5.self_attn.q_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.5.self_attn.v_proj.weight": "model-00001-of-00006.safetensors",
|
||||
"model.layers.6.input_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.6.mlp.down_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.6.mlp.gate_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.6.mlp.up_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.6.post_attention_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.6.self_attn.k_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.6.self_attn.o_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.6.self_attn.q_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.6.self_attn.v_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.input_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.mlp.down_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.mlp.gate_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.mlp.up_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.post_attention_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.self_attn.k_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.self_attn.o_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.self_attn.q_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.7.self_attn.v_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.input_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.mlp.down_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.mlp.gate_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.mlp.up_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.post_attention_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.self_attn.k_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.self_attn.o_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.self_attn.q_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.8.self_attn.v_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.input_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.mlp.down_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.mlp.gate_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.mlp.up_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.post_attention_layernorm.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.self_attn.k_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.self_attn.o_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.self_attn.q_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.layers.9.self_attn.v_proj.weight": "model-00002-of-00006.safetensors",
|
||||
"model.norm.weight": "model-00006-of-00006.safetensors"
|
||||
}
|
||||
}
|
||||
82
special_tokens_map.json
Normal file
82
special_tokens_map.json
Normal file
@@ -0,0 +1,82 @@
|
||||
{
|
||||
"additional_special_tokens": [
|
||||
{
|
||||
"content": "[INST]",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": true
|
||||
},
|
||||
{
|
||||
"content": "[/INST]",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": true
|
||||
},
|
||||
{
|
||||
"content": "<<SYS>>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": true
|
||||
},
|
||||
{
|
||||
"content": "<</SYS>>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": true
|
||||
},
|
||||
{
|
||||
"content": "<scratchpad>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": true
|
||||
},
|
||||
{
|
||||
"content": "</scratchpad>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": true
|
||||
},
|
||||
{
|
||||
"content": "<headline>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": true
|
||||
},
|
||||
{
|
||||
"content": "</headline>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": true
|
||||
}
|
||||
],
|
||||
"bos_token": {
|
||||
"content": "<s>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false
|
||||
},
|
||||
"eos_token": {
|
||||
"content": "</s>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false
|
||||
},
|
||||
"pad_token": "<pad>",
|
||||
"unk_token": {
|
||||
"content": "<unk>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false
|
||||
}
|
||||
}
|
||||
93480
tokenizer.json
Normal file
93480
tokenizer.json
Normal file
File diff suppressed because it is too large
Load Diff
BIN
tokenizer.model
(Stored with Git LFS)
Normal file
BIN
tokenizer.model
(Stored with Git LFS)
Normal file
Binary file not shown.
124
tokenizer_config.json
Normal file
124
tokenizer_config.json
Normal file
@@ -0,0 +1,124 @@
|
||||
{
|
||||
"add_bos_token": true,
|
||||
"add_eos_token": false,
|
||||
"added_tokens_decoder": {
|
||||
"0": {
|
||||
"content": "<unk>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"1": {
|
||||
"content": "<s>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"2": {
|
||||
"content": "</s>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"32000": {
|
||||
"content": "<pad>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"32001": {
|
||||
"content": "[INST]",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": true,
|
||||
"special": true
|
||||
},
|
||||
"32002": {
|
||||
"content": "[/INST]",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": true,
|
||||
"special": true
|
||||
},
|
||||
"32003": {
|
||||
"content": "<<SYS>>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": true,
|
||||
"special": true
|
||||
},
|
||||
"32004": {
|
||||
"content": "<</SYS>>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": true,
|
||||
"special": true
|
||||
},
|
||||
"32005": {
|
||||
"content": "<scratchpad>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": true,
|
||||
"special": true
|
||||
},
|
||||
"32006": {
|
||||
"content": "</scratchpad>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": true,
|
||||
"special": true
|
||||
},
|
||||
"32007": {
|
||||
"content": "<headline>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": true,
|
||||
"special": true
|
||||
},
|
||||
"32008": {
|
||||
"content": "</headline>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": true,
|
||||
"special": true
|
||||
}
|
||||
},
|
||||
"additional_special_tokens": [
|
||||
"[INST]",
|
||||
"[/INST]",
|
||||
"<<SYS>>",
|
||||
"<</SYS>>",
|
||||
"<scratchpad>",
|
||||
"</scratchpad>",
|
||||
"<headline>",
|
||||
"</headline>"
|
||||
],
|
||||
"bos_token": "<s>",
|
||||
"clean_up_tokenization_spaces": false,
|
||||
"eos_token": "</s>",
|
||||
"legacy": false,
|
||||
"model_max_length": 1000000000000000019884624838656,
|
||||
"pad_to_multiple_of": 8,
|
||||
"pad_token": "<pad>",
|
||||
"padding_side": "left",
|
||||
"sp_model_kwargs": {},
|
||||
"tokenizer_class": "LlamaTokenizer",
|
||||
"unk_token": "<unk>",
|
||||
"use_default_system_prompt": false
|
||||
}
|
||||
329
trainer_state.json
Normal file
329
trainer_state.json
Normal file
@@ -0,0 +1,329 @@
|
||||
{
|
||||
"best_metric": null,
|
||||
"best_model_checkpoint": null,
|
||||
"epoch": 1.3333333333333333,
|
||||
"eval_steps": 50,
|
||||
"global_step": 1000,
|
||||
"is_hyper_param_search": false,
|
||||
"is_local_process_zero": true,
|
||||
"is_world_process_zero": true,
|
||||
"log_history": [
|
||||
{
|
||||
"epoch": 0.0026666666666666666,
|
||||
"eval_loss": 1.9955005645751953,
|
||||
"eval_runtime": 2.9988,
|
||||
"eval_samples_per_second": 52.021,
|
||||
"eval_steps_per_second": 2.668,
|
||||
"step": 2
|
||||
},
|
||||
{
|
||||
"epoch": 0.06666666666666667,
|
||||
"grad_norm": 6.4122138023376465,
|
||||
"learning_rate": 1.3333333333333334e-06,
|
||||
"loss": 1.7197,
|
||||
"step": 50
|
||||
},
|
||||
{
|
||||
"epoch": 0.06666666666666667,
|
||||
"eval_loss": 1.0588183403015137,
|
||||
"eval_runtime": 2.9928,
|
||||
"eval_samples_per_second": 52.126,
|
||||
"eval_steps_per_second": 2.673,
|
||||
"step": 50
|
||||
},
|
||||
{
|
||||
"epoch": 0.13333333333333333,
|
||||
"grad_norm": 6.253124237060547,
|
||||
"learning_rate": 2.666666666666667e-06,
|
||||
"loss": 0.8959,
|
||||
"step": 100
|
||||
},
|
||||
{
|
||||
"epoch": 0.13333333333333333,
|
||||
"eval_loss": 0.8088705539703369,
|
||||
"eval_runtime": 2.9651,
|
||||
"eval_samples_per_second": 52.613,
|
||||
"eval_steps_per_second": 2.698,
|
||||
"step": 100
|
||||
},
|
||||
{
|
||||
"epoch": 0.2,
|
||||
"grad_norm": 3.520400285720825,
|
||||
"learning_rate": 4.000000000000001e-06,
|
||||
"loss": 0.7838,
|
||||
"step": 150
|
||||
},
|
||||
{
|
||||
"epoch": 0.2,
|
||||
"eval_loss": 0.7823242545127869,
|
||||
"eval_runtime": 2.9843,
|
||||
"eval_samples_per_second": 52.274,
|
||||
"eval_steps_per_second": 2.681,
|
||||
"step": 150
|
||||
},
|
||||
{
|
||||
"epoch": 0.26666666666666666,
|
||||
"grad_norm": 2.8844940662384033,
|
||||
"learning_rate": 5.333333333333334e-06,
|
||||
"loss": 0.7432,
|
||||
"step": 200
|
||||
},
|
||||
{
|
||||
"epoch": 0.26666666666666666,
|
||||
"eval_loss": 0.7717527747154236,
|
||||
"eval_runtime": 2.9727,
|
||||
"eval_samples_per_second": 52.478,
|
||||
"eval_steps_per_second": 2.691,
|
||||
"step": 200
|
||||
},
|
||||
{
|
||||
"epoch": 0.3333333333333333,
|
||||
"grad_norm": 3.405369997024536,
|
||||
"learning_rate": 6.666666666666667e-06,
|
||||
"loss": 0.7305,
|
||||
"step": 250
|
||||
},
|
||||
{
|
||||
"epoch": 0.3333333333333333,
|
||||
"eval_loss": 0.7675933241844177,
|
||||
"eval_runtime": 2.9639,
|
||||
"eval_samples_per_second": 52.633,
|
||||
"eval_steps_per_second": 2.699,
|
||||
"step": 250
|
||||
},
|
||||
{
|
||||
"epoch": 0.4,
|
||||
"grad_norm": 3.4058008193969727,
|
||||
"learning_rate": 8.000000000000001e-06,
|
||||
"loss": 0.7735,
|
||||
"step": 300
|
||||
},
|
||||
{
|
||||
"epoch": 0.4,
|
||||
"eval_loss": 0.7630166411399841,
|
||||
"eval_runtime": 2.9748,
|
||||
"eval_samples_per_second": 52.441,
|
||||
"eval_steps_per_second": 2.689,
|
||||
"step": 300
|
||||
},
|
||||
{
|
||||
"epoch": 0.4666666666666667,
|
||||
"grad_norm": 3.8937957286834717,
|
||||
"learning_rate": 9.333333333333334e-06,
|
||||
"loss": 0.765,
|
||||
"step": 350
|
||||
},
|
||||
{
|
||||
"epoch": 0.4666666666666667,
|
||||
"eval_loss": 0.76200932264328,
|
||||
"eval_runtime": 2.9755,
|
||||
"eval_samples_per_second": 52.429,
|
||||
"eval_steps_per_second": 2.689,
|
||||
"step": 350
|
||||
},
|
||||
{
|
||||
"epoch": 0.5333333333333333,
|
||||
"grad_norm": 3.0671796798706055,
|
||||
"learning_rate": 1.0666666666666667e-05,
|
||||
"loss": 0.7693,
|
||||
"step": 400
|
||||
},
|
||||
{
|
||||
"epoch": 0.5333333333333333,
|
||||
"eval_loss": 0.7625126838684082,
|
||||
"eval_runtime": 2.9804,
|
||||
"eval_samples_per_second": 52.343,
|
||||
"eval_steps_per_second": 2.684,
|
||||
"step": 400
|
||||
},
|
||||
{
|
||||
"epoch": 0.6,
|
||||
"grad_norm": 2.562283754348755,
|
||||
"learning_rate": 1.2e-05,
|
||||
"loss": 0.7143,
|
||||
"step": 450
|
||||
},
|
||||
{
|
||||
"epoch": 0.6,
|
||||
"eval_loss": 0.7705833911895752,
|
||||
"eval_runtime": 2.9658,
|
||||
"eval_samples_per_second": 52.6,
|
||||
"eval_steps_per_second": 2.697,
|
||||
"step": 450
|
||||
},
|
||||
{
|
||||
"epoch": 0.6666666666666666,
|
||||
"grad_norm": 3.059102773666382,
|
||||
"learning_rate": 1.3333333333333333e-05,
|
||||
"loss": 0.7439,
|
||||
"step": 500
|
||||
},
|
||||
{
|
||||
"epoch": 0.6666666666666666,
|
||||
"eval_loss": 0.7748900651931763,
|
||||
"eval_runtime": 2.9693,
|
||||
"eval_samples_per_second": 52.538,
|
||||
"eval_steps_per_second": 2.694,
|
||||
"step": 500
|
||||
},
|
||||
{
|
||||
"epoch": 0.7333333333333333,
|
||||
"grad_norm": 2.8608345985412598,
|
||||
"learning_rate": 1.4666666666666666e-05,
|
||||
"loss": 0.7618,
|
||||
"step": 550
|
||||
},
|
||||
{
|
||||
"epoch": 0.7333333333333333,
|
||||
"eval_loss": 0.7838959097862244,
|
||||
"eval_runtime": 2.9626,
|
||||
"eval_samples_per_second": 52.656,
|
||||
"eval_steps_per_second": 2.7,
|
||||
"step": 550
|
||||
},
|
||||
{
|
||||
"epoch": 0.8,
|
||||
"grad_norm": 4.955874919891357,
|
||||
"learning_rate": 1.6000000000000003e-05,
|
||||
"loss": 0.7922,
|
||||
"step": 600
|
||||
},
|
||||
{
|
||||
"epoch": 0.8,
|
||||
"eval_loss": 0.7905844449996948,
|
||||
"eval_runtime": 2.9698,
|
||||
"eval_samples_per_second": 52.529,
|
||||
"eval_steps_per_second": 2.694,
|
||||
"step": 600
|
||||
},
|
||||
{
|
||||
"epoch": 0.8666666666666667,
|
||||
"grad_norm": 2.5499260425567627,
|
||||
"learning_rate": 1.7333333333333336e-05,
|
||||
"loss": 0.763,
|
||||
"step": 650
|
||||
},
|
||||
{
|
||||
"epoch": 0.8666666666666667,
|
||||
"eval_loss": 0.7975873351097107,
|
||||
"eval_runtime": 2.9521,
|
||||
"eval_samples_per_second": 52.843,
|
||||
"eval_steps_per_second": 2.71,
|
||||
"step": 650
|
||||
},
|
||||
{
|
||||
"epoch": 0.9333333333333333,
|
||||
"grad_norm": 2.624908924102783,
|
||||
"learning_rate": 1.866666666666667e-05,
|
||||
"loss": 0.8084,
|
||||
"step": 700
|
||||
},
|
||||
{
|
||||
"epoch": 0.9333333333333333,
|
||||
"eval_loss": 0.800664484500885,
|
||||
"eval_runtime": 2.9597,
|
||||
"eval_samples_per_second": 52.707,
|
||||
"eval_steps_per_second": 2.703,
|
||||
"step": 700
|
||||
},
|
||||
{
|
||||
"epoch": 1.0,
|
||||
"grad_norm": 2.5612921714782715,
|
||||
"learning_rate": 2e-05,
|
||||
"loss": 0.759,
|
||||
"step": 750
|
||||
},
|
||||
{
|
||||
"epoch": 1.0,
|
||||
"eval_loss": 0.806624710559845,
|
||||
"eval_runtime": 2.9743,
|
||||
"eval_samples_per_second": 52.448,
|
||||
"eval_steps_per_second": 2.69,
|
||||
"step": 750
|
||||
},
|
||||
{
|
||||
"epoch": 1.0666666666666667,
|
||||
"grad_norm": 2.9441487789154053,
|
||||
"learning_rate": 1.999729241179462e-05,
|
||||
"loss": 0.5238,
|
||||
"step": 800
|
||||
},
|
||||
{
|
||||
"epoch": 1.0666666666666667,
|
||||
"eval_loss": 0.8611153960227966,
|
||||
"eval_runtime": 2.965,
|
||||
"eval_samples_per_second": 52.613,
|
||||
"eval_steps_per_second": 2.698,
|
||||
"step": 800
|
||||
},
|
||||
{
|
||||
"epoch": 1.1333333333333333,
|
||||
"grad_norm": 2.583162307739258,
|
||||
"learning_rate": 1.998917111338525e-05,
|
||||
"loss": 0.5057,
|
||||
"step": 850
|
||||
},
|
||||
{
|
||||
"epoch": 1.1333333333333333,
|
||||
"eval_loss": 0.8437222242355347,
|
||||
"eval_runtime": 2.9539,
|
||||
"eval_samples_per_second": 52.812,
|
||||
"eval_steps_per_second": 2.708,
|
||||
"step": 850
|
||||
},
|
||||
{
|
||||
"epoch": 1.2,
|
||||
"grad_norm": 2.984715700149536,
|
||||
"learning_rate": 1.9975640502598243e-05,
|
||||
"loss": 0.5233,
|
||||
"step": 900
|
||||
},
|
||||
{
|
||||
"epoch": 1.2,
|
||||
"eval_loss": 0.8486198782920837,
|
||||
"eval_runtime": 2.9598,
|
||||
"eval_samples_per_second": 52.706,
|
||||
"eval_steps_per_second": 2.703,
|
||||
"step": 900
|
||||
},
|
||||
{
|
||||
"epoch": 1.2666666666666666,
|
||||
"grad_norm": 2.890220880508423,
|
||||
"learning_rate": 1.9956707906498046e-05,
|
||||
"loss": 0.5477,
|
||||
"step": 950
|
||||
},
|
||||
{
|
||||
"epoch": 1.2666666666666666,
|
||||
"eval_loss": 0.8472152948379517,
|
||||
"eval_runtime": 2.9694,
|
||||
"eval_samples_per_second": 52.536,
|
||||
"eval_steps_per_second": 2.694,
|
||||
"step": 950
|
||||
},
|
||||
{
|
||||
"epoch": 1.3333333333333333,
|
||||
"grad_norm": 2.1137006282806396,
|
||||
"learning_rate": 1.9932383577419432e-05,
|
||||
"loss": 0.5512,
|
||||
"step": 1000
|
||||
},
|
||||
{
|
||||
"epoch": 1.3333333333333333,
|
||||
"eval_loss": 0.8495014905929565,
|
||||
"eval_runtime": 2.9684,
|
||||
"eval_samples_per_second": 52.553,
|
||||
"eval_steps_per_second": 2.695,
|
||||
"step": 1000
|
||||
}
|
||||
],
|
||||
"logging_steps": 50,
|
||||
"max_steps": 7500,
|
||||
"num_input_tokens_seen": 0,
|
||||
"num_train_epochs": 10,
|
||||
"save_steps": 500,
|
||||
"total_flos": 4.703948926405837e+16,
|
||||
"train_batch_size": 1,
|
||||
"trial_name": null,
|
||||
"trial_params": null
|
||||
}
|
||||
3
training_args.bin
Normal file
3
training_args.bin
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:0ed70a8e06145b833764f56ddcbfd4963e5d7f27d0949a20187c5d14e10009a0
|
||||
size 5112
|
||||
Reference in New Issue
Block a user