tianyudu commited on
Commit
718a65d
·
verified ·
1 Parent(s): da1493d

Upload folder using huggingface_hub

Browse files
ft_13b_NLSY79_with_birth_year_ckpt_3/config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "togethercomputer/llama-2-13b",
3
+ "architectures": [
4
+ "LlamaForCausalLM"
5
+ ],
6
+ "attention_bias": false,
7
+ "attention_dropout": 0.0,
8
+ "bos_token_id": 1,
9
+ "eos_token_id": 2,
10
+ "head_dim": 128,
11
+ "hidden_act": "silu",
12
+ "hidden_size": 5120,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 13824,
15
+ "max_position_embeddings": 4096,
16
+ "mlp_bias": false,
17
+ "model_type": "llama",
18
+ "num_attention_heads": 40,
19
+ "num_hidden_layers": 40,
20
+ "num_key_value_heads": 40,
21
+ "pad_token_id": 0,
22
+ "pretraining_tp": 2,
23
+ "rms_norm_eps": 1e-05,
24
+ "rope_scaling": null,
25
+ "rope_theta": 10000.0,
26
+ "tie_word_embeddings": false,
27
+ "torch_dtype": "float32",
28
+ "transformers_version": "4.45.1",
29
+ "use_cache": false,
30
+ "vocab_size": 32000
31
+ }
ft_13b_NLSY79_with_birth_year_ckpt_3/generation_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 1,
3
+ "do_sample": true,
4
+ "eos_token_id": 2,
5
+ "max_length": 4096,
6
+ "pad_token_id": 0,
7
+ "temperature": 0.6,
8
+ "top_p": 0.9,
9
+ "transformers_version": "4.45.1"
10
+ }
ft_13b_NLSY79_with_birth_year_ckpt_3/model-00001-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:80fc9efbb305ed7bee48f382d39f8381939a6bee6298590ba86e6e1889015da9
3
+ size 4881247856
ft_13b_NLSY79_with_birth_year_ckpt_3/model-00002-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c7a9140b4aa42a813911cfc134e42534a31cb08a4ff849267f78d6a4a80cd0b8
3
+ size 4970418112
ft_13b_NLSY79_with_birth_year_ckpt_3/model-00003-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:af7fd08dcfaf2579ba153e51c2c675611b65cd9fe644d8ff3b1c62f8341ca8b6
3
+ size 4970418120
ft_13b_NLSY79_with_birth_year_ckpt_3/model-00004-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50d227d8549c40b18014199eb42baa478ffdfe923e94507f1dcd1daf8975fa87
3
+ size 4970418144
ft_13b_NLSY79_with_birth_year_ckpt_3/model-00005-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:999e4d4373bb8718fcb6d4890d69a6d349c46c4e2e1a607fca3a63a566289a99
3
+ size 4970418144
ft_13b_NLSY79_with_birth_year_ckpt_3/model-00006-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:145b54b90b462ddae21b4f88771b8eabbeccaa1cec899d676cdcf12cd82015fe
3
+ size 4792119040
ft_13b_NLSY79_with_birth_year_ckpt_3/model-00007-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e4a92d2f2bef6093c58e2ad383d79602439b404b3ff44649e08a8e6893924384
3
+ size 4792160232
ft_13b_NLSY79_with_birth_year_ckpt_3/model-00008-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1903719bcdf36504e73e7aabbe86ced67735cbe4bcc9a814dbc9e2af4148847a
3
+ size 4792160224
ft_13b_NLSY79_with_birth_year_ckpt_3/model-00009-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:37928acbe045a6687071ee7771ef6d2404488ec9d910db9bf847d4b81f58b0d2
3
+ size 4970418144
ft_13b_NLSY79_with_birth_year_ckpt_3/model-00010-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e10a2f2df781dc8b7f39b440beec3c6b88eb7ff7ccfaab11bd4ecc6eb4506a28
3
+ size 4970418144
ft_13b_NLSY79_with_birth_year_ckpt_3/model-00011-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:efffa86d04cdfb248e62d7255e2ee73d1ded60bfe06d7a17d0fe02403a623a1c
3
+ size 2983303184
ft_13b_NLSY79_with_birth_year_ckpt_3/model.safetensors.index.json ADDED
@@ -0,0 +1,370 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_size": 52063457280
4
+ },
5
+ "weight_map": {
6
+ "lm_head.weight": "model-00011-of-00011.safetensors",
7
+ "model.embed_tokens.weight": "model-00001-of-00011.safetensors",
8
+ "model.layers.0.input_layernorm.weight": "model-00001-of-00011.safetensors",
9
+ "model.layers.0.mlp.down_proj.weight": "model-00001-of-00011.safetensors",
10
+ "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00011.safetensors",
11
+ "model.layers.0.mlp.up_proj.weight": "model-00001-of-00011.safetensors",
12
+ "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00011.safetensors",
13
+ "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00011.safetensors",
14
+ "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00011.safetensors",
15
+ "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00011.safetensors",
16
+ "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00011.safetensors",
17
+ "model.layers.1.input_layernorm.weight": "model-00001-of-00011.safetensors",
18
+ "model.layers.1.mlp.down_proj.weight": "model-00001-of-00011.safetensors",
19
+ "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00011.safetensors",
20
+ "model.layers.1.mlp.up_proj.weight": "model-00001-of-00011.safetensors",
21
+ "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00011.safetensors",
22
+ "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00011.safetensors",
23
+ "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00011.safetensors",
24
+ "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00011.safetensors",
25
+ "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00011.safetensors",
26
+ "model.layers.10.input_layernorm.weight": "model-00003-of-00011.safetensors",
27
+ "model.layers.10.mlp.down_proj.weight": "model-00003-of-00011.safetensors",
28
+ "model.layers.10.mlp.gate_proj.weight": "model-00003-of-00011.safetensors",
29
+ "model.layers.10.mlp.up_proj.weight": "model-00003-of-00011.safetensors",
30
+ "model.layers.10.post_attention_layernorm.weight": "model-00003-of-00011.safetensors",
31
+ "model.layers.10.self_attn.k_proj.weight": "model-00003-of-00011.safetensors",
32
+ "model.layers.10.self_attn.o_proj.weight": "model-00003-of-00011.safetensors",
33
+ "model.layers.10.self_attn.q_proj.weight": "model-00003-of-00011.safetensors",
34
+ "model.layers.10.self_attn.v_proj.weight": "model-00003-of-00011.safetensors",
35
+ "model.layers.11.input_layernorm.weight": "model-00004-of-00011.safetensors",
36
+ "model.layers.11.mlp.down_proj.weight": "model-00004-of-00011.safetensors",
37
+ "model.layers.11.mlp.gate_proj.weight": "model-00004-of-00011.safetensors",
38
+ "model.layers.11.mlp.up_proj.weight": "model-00004-of-00011.safetensors",
39
+ "model.layers.11.post_attention_layernorm.weight": "model-00004-of-00011.safetensors",
40
+ "model.layers.11.self_attn.k_proj.weight": "model-00003-of-00011.safetensors",
41
+ "model.layers.11.self_attn.o_proj.weight": "model-00004-of-00011.safetensors",
42
+ "model.layers.11.self_attn.q_proj.weight": "model-00003-of-00011.safetensors",
43
+ "model.layers.11.self_attn.v_proj.weight": "model-00004-of-00011.safetensors",
44
+ "model.layers.12.input_layernorm.weight": "model-00004-of-00011.safetensors",
45
+ "model.layers.12.mlp.down_proj.weight": "model-00004-of-00011.safetensors",
46
+ "model.layers.12.mlp.gate_proj.weight": "model-00004-of-00011.safetensors",
47
+ "model.layers.12.mlp.up_proj.weight": "model-00004-of-00011.safetensors",
48
+ "model.layers.12.post_attention_layernorm.weight": "model-00004-of-00011.safetensors",
49
+ "model.layers.12.self_attn.k_proj.weight": "model-00004-of-00011.safetensors",
50
+ "model.layers.12.self_attn.o_proj.weight": "model-00004-of-00011.safetensors",
51
+ "model.layers.12.self_attn.q_proj.weight": "model-00004-of-00011.safetensors",
52
+ "model.layers.12.self_attn.v_proj.weight": "model-00004-of-00011.safetensors",
53
+ "model.layers.13.input_layernorm.weight": "model-00004-of-00011.safetensors",
54
+ "model.layers.13.mlp.down_proj.weight": "model-00004-of-00011.safetensors",
55
+ "model.layers.13.mlp.gate_proj.weight": "model-00004-of-00011.safetensors",
56
+ "model.layers.13.mlp.up_proj.weight": "model-00004-of-00011.safetensors",
57
+ "model.layers.13.post_attention_layernorm.weight": "model-00004-of-00011.safetensors",
58
+ "model.layers.13.self_attn.k_proj.weight": "model-00004-of-00011.safetensors",
59
+ "model.layers.13.self_attn.o_proj.weight": "model-00004-of-00011.safetensors",
60
+ "model.layers.13.self_attn.q_proj.weight": "model-00004-of-00011.safetensors",
61
+ "model.layers.13.self_attn.v_proj.weight": "model-00004-of-00011.safetensors",
62
+ "model.layers.14.input_layernorm.weight": "model-00004-of-00011.safetensors",
63
+ "model.layers.14.mlp.down_proj.weight": "model-00004-of-00011.safetensors",
64
+ "model.layers.14.mlp.gate_proj.weight": "model-00004-of-00011.safetensors",
65
+ "model.layers.14.mlp.up_proj.weight": "model-00004-of-00011.safetensors",
66
+ "model.layers.14.post_attention_layernorm.weight": "model-00004-of-00011.safetensors",
67
+ "model.layers.14.self_attn.k_proj.weight": "model-00004-of-00011.safetensors",
68
+ "model.layers.14.self_attn.o_proj.weight": "model-00004-of-00011.safetensors",
69
+ "model.layers.14.self_attn.q_proj.weight": "model-00004-of-00011.safetensors",
70
+ "model.layers.14.self_attn.v_proj.weight": "model-00004-of-00011.safetensors",
71
+ "model.layers.15.input_layernorm.weight": "model-00005-of-00011.safetensors",
72
+ "model.layers.15.mlp.down_proj.weight": "model-00005-of-00011.safetensors",
73
+ "model.layers.15.mlp.gate_proj.weight": "model-00005-of-00011.safetensors",
74
+ "model.layers.15.mlp.up_proj.weight": "model-00005-of-00011.safetensors",
75
+ "model.layers.15.post_attention_layernorm.weight": "model-00005-of-00011.safetensors",
76
+ "model.layers.15.self_attn.k_proj.weight": "model-00005-of-00011.safetensors",
77
+ "model.layers.15.self_attn.o_proj.weight": "model-00005-of-00011.safetensors",
78
+ "model.layers.15.self_attn.q_proj.weight": "model-00004-of-00011.safetensors",
79
+ "model.layers.15.self_attn.v_proj.weight": "model-00005-of-00011.safetensors",
80
+ "model.layers.16.input_layernorm.weight": "model-00005-of-00011.safetensors",
81
+ "model.layers.16.mlp.down_proj.weight": "model-00005-of-00011.safetensors",
82
+ "model.layers.16.mlp.gate_proj.weight": "model-00005-of-00011.safetensors",
83
+ "model.layers.16.mlp.up_proj.weight": "model-00005-of-00011.safetensors",
84
+ "model.layers.16.post_attention_layernorm.weight": "model-00005-of-00011.safetensors",
85
+ "model.layers.16.self_attn.k_proj.weight": "model-00005-of-00011.safetensors",
86
+ "model.layers.16.self_attn.o_proj.weight": "model-00005-of-00011.safetensors",
87
+ "model.layers.16.self_attn.q_proj.weight": "model-00005-of-00011.safetensors",
88
+ "model.layers.16.self_attn.v_proj.weight": "model-00005-of-00011.safetensors",
89
+ "model.layers.17.input_layernorm.weight": "model-00005-of-00011.safetensors",
90
+ "model.layers.17.mlp.down_proj.weight": "model-00005-of-00011.safetensors",
91
+ "model.layers.17.mlp.gate_proj.weight": "model-00005-of-00011.safetensors",
92
+ "model.layers.17.mlp.up_proj.weight": "model-00005-of-00011.safetensors",
93
+ "model.layers.17.post_attention_layernorm.weight": "model-00005-of-00011.safetensors",
94
+ "model.layers.17.self_attn.k_proj.weight": "model-00005-of-00011.safetensors",
95
+ "model.layers.17.self_attn.o_proj.weight": "model-00005-of-00011.safetensors",
96
+ "model.layers.17.self_attn.q_proj.weight": "model-00005-of-00011.safetensors",
97
+ "model.layers.17.self_attn.v_proj.weight": "model-00005-of-00011.safetensors",
98
+ "model.layers.18.input_layernorm.weight": "model-00005-of-00011.safetensors",
99
+ "model.layers.18.mlp.down_proj.weight": "model-00005-of-00011.safetensors",
100
+ "model.layers.18.mlp.gate_proj.weight": "model-00005-of-00011.safetensors",
101
+ "model.layers.18.mlp.up_proj.weight": "model-00005-of-00011.safetensors",
102
+ "model.layers.18.post_attention_layernorm.weight": "model-00005-of-00011.safetensors",
103
+ "model.layers.18.self_attn.k_proj.weight": "model-00005-of-00011.safetensors",
104
+ "model.layers.18.self_attn.o_proj.weight": "model-00005-of-00011.safetensors",
105
+ "model.layers.18.self_attn.q_proj.weight": "model-00005-of-00011.safetensors",
106
+ "model.layers.18.self_attn.v_proj.weight": "model-00005-of-00011.safetensors",
107
+ "model.layers.19.input_layernorm.weight": "model-00006-of-00011.safetensors",
108
+ "model.layers.19.mlp.down_proj.weight": "model-00006-of-00011.safetensors",
109
+ "model.layers.19.mlp.gate_proj.weight": "model-00006-of-00011.safetensors",
110
+ "model.layers.19.mlp.up_proj.weight": "model-00006-of-00011.safetensors",
111
+ "model.layers.19.post_attention_layernorm.weight": "model-00006-of-00011.safetensors",
112
+ "model.layers.19.self_attn.k_proj.weight": "model-00006-of-00011.safetensors",
113
+ "model.layers.19.self_attn.o_proj.weight": "model-00006-of-00011.safetensors",
114
+ "model.layers.19.self_attn.q_proj.weight": "model-00006-of-00011.safetensors",
115
+ "model.layers.19.self_attn.v_proj.weight": "model-00006-of-00011.safetensors",
116
+ "model.layers.2.input_layernorm.weight": "model-00001-of-00011.safetensors",
117
+ "model.layers.2.mlp.down_proj.weight": "model-00001-of-00011.safetensors",
118
+ "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00011.safetensors",
119
+ "model.layers.2.mlp.up_proj.weight": "model-00001-of-00011.safetensors",
120
+ "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00011.safetensors",
121
+ "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00011.safetensors",
122
+ "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00011.safetensors",
123
+ "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00011.safetensors",
124
+ "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00011.safetensors",
125
+ "model.layers.20.input_layernorm.weight": "model-00006-of-00011.safetensors",
126
+ "model.layers.20.mlp.down_proj.weight": "model-00006-of-00011.safetensors",
127
+ "model.layers.20.mlp.gate_proj.weight": "model-00006-of-00011.safetensors",
128
+ "model.layers.20.mlp.up_proj.weight": "model-00006-of-00011.safetensors",
129
+ "model.layers.20.post_attention_layernorm.weight": "model-00006-of-00011.safetensors",
130
+ "model.layers.20.self_attn.k_proj.weight": "model-00006-of-00011.safetensors",
131
+ "model.layers.20.self_attn.o_proj.weight": "model-00006-of-00011.safetensors",
132
+ "model.layers.20.self_attn.q_proj.weight": "model-00006-of-00011.safetensors",
133
+ "model.layers.20.self_attn.v_proj.weight": "model-00006-of-00011.safetensors",
134
+ "model.layers.21.input_layernorm.weight": "model-00006-of-00011.safetensors",
135
+ "model.layers.21.mlp.down_proj.weight": "model-00006-of-00011.safetensors",
136
+ "model.layers.21.mlp.gate_proj.weight": "model-00006-of-00011.safetensors",
137
+ "model.layers.21.mlp.up_proj.weight": "model-00006-of-00011.safetensors",
138
+ "model.layers.21.post_attention_layernorm.weight": "model-00006-of-00011.safetensors",
139
+ "model.layers.21.self_attn.k_proj.weight": "model-00006-of-00011.safetensors",
140
+ "model.layers.21.self_attn.o_proj.weight": "model-00006-of-00011.safetensors",
141
+ "model.layers.21.self_attn.q_proj.weight": "model-00006-of-00011.safetensors",
142
+ "model.layers.21.self_attn.v_proj.weight": "model-00006-of-00011.safetensors",
143
+ "model.layers.22.input_layernorm.weight": "model-00007-of-00011.safetensors",
144
+ "model.layers.22.mlp.down_proj.weight": "model-00007-of-00011.safetensors",
145
+ "model.layers.22.mlp.gate_proj.weight": "model-00006-of-00011.safetensors",
146
+ "model.layers.22.mlp.up_proj.weight": "model-00006-of-00011.safetensors",
147
+ "model.layers.22.post_attention_layernorm.weight": "model-00007-of-00011.safetensors",
148
+ "model.layers.22.self_attn.k_proj.weight": "model-00006-of-00011.safetensors",
149
+ "model.layers.22.self_attn.o_proj.weight": "model-00006-of-00011.safetensors",
150
+ "model.layers.22.self_attn.q_proj.weight": "model-00006-of-00011.safetensors",
151
+ "model.layers.22.self_attn.v_proj.weight": "model-00006-of-00011.safetensors",
152
+ "model.layers.23.input_layernorm.weight": "model-00007-of-00011.safetensors",
153
+ "model.layers.23.mlp.down_proj.weight": "model-00007-of-00011.safetensors",
154
+ "model.layers.23.mlp.gate_proj.weight": "model-00007-of-00011.safetensors",
155
+ "model.layers.23.mlp.up_proj.weight": "model-00007-of-00011.safetensors",
156
+ "model.layers.23.post_attention_layernorm.weight": "model-00007-of-00011.safetensors",
157
+ "model.layers.23.self_attn.k_proj.weight": "model-00007-of-00011.safetensors",
158
+ "model.layers.23.self_attn.o_proj.weight": "model-00007-of-00011.safetensors",
159
+ "model.layers.23.self_attn.q_proj.weight": "model-00007-of-00011.safetensors",
160
+ "model.layers.23.self_attn.v_proj.weight": "model-00007-of-00011.safetensors",
161
+ "model.layers.24.input_layernorm.weight": "model-00007-of-00011.safetensors",
162
+ "model.layers.24.mlp.down_proj.weight": "model-00007-of-00011.safetensors",
163
+ "model.layers.24.mlp.gate_proj.weight": "model-00007-of-00011.safetensors",
164
+ "model.layers.24.mlp.up_proj.weight": "model-00007-of-00011.safetensors",
165
+ "model.layers.24.post_attention_layernorm.weight": "model-00007-of-00011.safetensors",
166
+ "model.layers.24.self_attn.k_proj.weight": "model-00007-of-00011.safetensors",
167
+ "model.layers.24.self_attn.o_proj.weight": "model-00007-of-00011.safetensors",
168
+ "model.layers.24.self_attn.q_proj.weight": "model-00007-of-00011.safetensors",
169
+ "model.layers.24.self_attn.v_proj.weight": "model-00007-of-00011.safetensors",
170
+ "model.layers.25.input_layernorm.weight": "model-00007-of-00011.safetensors",
171
+ "model.layers.25.mlp.down_proj.weight": "model-00007-of-00011.safetensors",
172
+ "model.layers.25.mlp.gate_proj.weight": "model-00007-of-00011.safetensors",
173
+ "model.layers.25.mlp.up_proj.weight": "model-00007-of-00011.safetensors",
174
+ "model.layers.25.post_attention_layernorm.weight": "model-00007-of-00011.safetensors",
175
+ "model.layers.25.self_attn.k_proj.weight": "model-00007-of-00011.safetensors",
176
+ "model.layers.25.self_attn.o_proj.weight": "model-00007-of-00011.safetensors",
177
+ "model.layers.25.self_attn.q_proj.weight": "model-00007-of-00011.safetensors",
178
+ "model.layers.25.self_attn.v_proj.weight": "model-00007-of-00011.safetensors",
179
+ "model.layers.26.input_layernorm.weight": "model-00008-of-00011.safetensors",
180
+ "model.layers.26.mlp.down_proj.weight": "model-00008-of-00011.safetensors",
181
+ "model.layers.26.mlp.gate_proj.weight": "model-00007-of-00011.safetensors",
182
+ "model.layers.26.mlp.up_proj.weight": "model-00008-of-00011.safetensors",
183
+ "model.layers.26.post_attention_layernorm.weight": "model-00008-of-00011.safetensors",
184
+ "model.layers.26.self_attn.k_proj.weight": "model-00007-of-00011.safetensors",
185
+ "model.layers.26.self_attn.o_proj.weight": "model-00007-of-00011.safetensors",
186
+ "model.layers.26.self_attn.q_proj.weight": "model-00007-of-00011.safetensors",
187
+ "model.layers.26.self_attn.v_proj.weight": "model-00007-of-00011.safetensors",
188
+ "model.layers.27.input_layernorm.weight": "model-00008-of-00011.safetensors",
189
+ "model.layers.27.mlp.down_proj.weight": "model-00008-of-00011.safetensors",
190
+ "model.layers.27.mlp.gate_proj.weight": "model-00008-of-00011.safetensors",
191
+ "model.layers.27.mlp.up_proj.weight": "model-00008-of-00011.safetensors",
192
+ "model.layers.27.post_attention_layernorm.weight": "model-00008-of-00011.safetensors",
193
+ "model.layers.27.self_attn.k_proj.weight": "model-00008-of-00011.safetensors",
194
+ "model.layers.27.self_attn.o_proj.weight": "model-00008-of-00011.safetensors",
195
+ "model.layers.27.self_attn.q_proj.weight": "model-00008-of-00011.safetensors",
196
+ "model.layers.27.self_attn.v_proj.weight": "model-00008-of-00011.safetensors",
197
+ "model.layers.28.input_layernorm.weight": "model-00008-of-00011.safetensors",
198
+ "model.layers.28.mlp.down_proj.weight": "model-00008-of-00011.safetensors",
199
+ "model.layers.28.mlp.gate_proj.weight": "model-00008-of-00011.safetensors",
200
+ "model.layers.28.mlp.up_proj.weight": "model-00008-of-00011.safetensors",
201
+ "model.layers.28.post_attention_layernorm.weight": "model-00008-of-00011.safetensors",
202
+ "model.layers.28.self_attn.k_proj.weight": "model-00008-of-00011.safetensors",
203
+ "model.layers.28.self_attn.o_proj.weight": "model-00008-of-00011.safetensors",
204
+ "model.layers.28.self_attn.q_proj.weight": "model-00008-of-00011.safetensors",
205
+ "model.layers.28.self_attn.v_proj.weight": "model-00008-of-00011.safetensors",
206
+ "model.layers.29.input_layernorm.weight": "model-00008-of-00011.safetensors",
207
+ "model.layers.29.mlp.down_proj.weight": "model-00008-of-00011.safetensors",
208
+ "model.layers.29.mlp.gate_proj.weight": "model-00008-of-00011.safetensors",
209
+ "model.layers.29.mlp.up_proj.weight": "model-00008-of-00011.safetensors",
210
+ "model.layers.29.post_attention_layernorm.weight": "model-00008-of-00011.safetensors",
211
+ "model.layers.29.self_attn.k_proj.weight": "model-00008-of-00011.safetensors",
212
+ "model.layers.29.self_attn.o_proj.weight": "model-00008-of-00011.safetensors",
213
+ "model.layers.29.self_attn.q_proj.weight": "model-00008-of-00011.safetensors",
214
+ "model.layers.29.self_attn.v_proj.weight": "model-00008-of-00011.safetensors",
215
+ "model.layers.3.input_layernorm.weight": "model-00002-of-00011.safetensors",
216
+ "model.layers.3.mlp.down_proj.weight": "model-00002-of-00011.safetensors",
217
+ "model.layers.3.mlp.gate_proj.weight": "model-00002-of-00011.safetensors",
218
+ "model.layers.3.mlp.up_proj.weight": "model-00002-of-00011.safetensors",
219
+ "model.layers.3.post_attention_layernorm.weight": "model-00002-of-00011.safetensors",
220
+ "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00011.safetensors",
221
+ "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00011.safetensors",
222
+ "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00011.safetensors",
223
+ "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00011.safetensors",
224
+ "model.layers.30.input_layernorm.weight": "model-00009-of-00011.safetensors",
225
+ "model.layers.30.mlp.down_proj.weight": "model-00009-of-00011.safetensors",
226
+ "model.layers.30.mlp.gate_proj.weight": "model-00009-of-00011.safetensors",
227
+ "model.layers.30.mlp.up_proj.weight": "model-00009-of-00011.safetensors",
228
+ "model.layers.30.post_attention_layernorm.weight": "model-00009-of-00011.safetensors",
229
+ "model.layers.30.self_attn.k_proj.weight": "model-00008-of-00011.safetensors",
230
+ "model.layers.30.self_attn.o_proj.weight": "model-00008-of-00011.safetensors",
231
+ "model.layers.30.self_attn.q_proj.weight": "model-00008-of-00011.safetensors",
232
+ "model.layers.30.self_attn.v_proj.weight": "model-00008-of-00011.safetensors",
233
+ "model.layers.31.input_layernorm.weight": "model-00009-of-00011.safetensors",
234
+ "model.layers.31.mlp.down_proj.weight": "model-00009-of-00011.safetensors",
235
+ "model.layers.31.mlp.gate_proj.weight": "model-00009-of-00011.safetensors",
236
+ "model.layers.31.mlp.up_proj.weight": "model-00009-of-00011.safetensors",
237
+ "model.layers.31.post_attention_layernorm.weight": "model-00009-of-00011.safetensors",
238
+ "model.layers.31.self_attn.k_proj.weight": "model-00009-of-00011.safetensors",
239
+ "model.layers.31.self_attn.o_proj.weight": "model-00009-of-00011.safetensors",
240
+ "model.layers.31.self_attn.q_proj.weight": "model-00009-of-00011.safetensors",
241
+ "model.layers.31.self_attn.v_proj.weight": "model-00009-of-00011.safetensors",
242
+ "model.layers.32.input_layernorm.weight": "model-00009-of-00011.safetensors",
243
+ "model.layers.32.mlp.down_proj.weight": "model-00009-of-00011.safetensors",
244
+ "model.layers.32.mlp.gate_proj.weight": "model-00009-of-00011.safetensors",
245
+ "model.layers.32.mlp.up_proj.weight": "model-00009-of-00011.safetensors",
246
+ "model.layers.32.post_attention_layernorm.weight": "model-00009-of-00011.safetensors",
247
+ "model.layers.32.self_attn.k_proj.weight": "model-00009-of-00011.safetensors",
248
+ "model.layers.32.self_attn.o_proj.weight": "model-00009-of-00011.safetensors",
249
+ "model.layers.32.self_attn.q_proj.weight": "model-00009-of-00011.safetensors",
250
+ "model.layers.32.self_attn.v_proj.weight": "model-00009-of-00011.safetensors",
251
+ "model.layers.33.input_layernorm.weight": "model-00009-of-00011.safetensors",
252
+ "model.layers.33.mlp.down_proj.weight": "model-00009-of-00011.safetensors",
253
+ "model.layers.33.mlp.gate_proj.weight": "model-00009-of-00011.safetensors",
254
+ "model.layers.33.mlp.up_proj.weight": "model-00009-of-00011.safetensors",
255
+ "model.layers.33.post_attention_layernorm.weight": "model-00009-of-00011.safetensors",
256
+ "model.layers.33.self_attn.k_proj.weight": "model-00009-of-00011.safetensors",
257
+ "model.layers.33.self_attn.o_proj.weight": "model-00009-of-00011.safetensors",
258
+ "model.layers.33.self_attn.q_proj.weight": "model-00009-of-00011.safetensors",
259
+ "model.layers.33.self_attn.v_proj.weight": "model-00009-of-00011.safetensors",
260
+ "model.layers.34.input_layernorm.weight": "model-00010-of-00011.safetensors",
261
+ "model.layers.34.mlp.down_proj.weight": "model-00010-of-00011.safetensors",
262
+ "model.layers.34.mlp.gate_proj.weight": "model-00010-of-00011.safetensors",
263
+ "model.layers.34.mlp.up_proj.weight": "model-00010-of-00011.safetensors",
264
+ "model.layers.34.post_attention_layernorm.weight": "model-00010-of-00011.safetensors",
265
+ "model.layers.34.self_attn.k_proj.weight": "model-00009-of-00011.safetensors",
266
+ "model.layers.34.self_attn.o_proj.weight": "model-00010-of-00011.safetensors",
267
+ "model.layers.34.self_attn.q_proj.weight": "model-00009-of-00011.safetensors",
268
+ "model.layers.34.self_attn.v_proj.weight": "model-00009-of-00011.safetensors",
269
+ "model.layers.35.input_layernorm.weight": "model-00010-of-00011.safetensors",
270
+ "model.layers.35.mlp.down_proj.weight": "model-00010-of-00011.safetensors",
271
+ "model.layers.35.mlp.gate_proj.weight": "model-00010-of-00011.safetensors",
272
+ "model.layers.35.mlp.up_proj.weight": "model-00010-of-00011.safetensors",
273
+ "model.layers.35.post_attention_layernorm.weight": "model-00010-of-00011.safetensors",
274
+ "model.layers.35.self_attn.k_proj.weight": "model-00010-of-00011.safetensors",
275
+ "model.layers.35.self_attn.o_proj.weight": "model-00010-of-00011.safetensors",
276
+ "model.layers.35.self_attn.q_proj.weight": "model-00010-of-00011.safetensors",
277
+ "model.layers.35.self_attn.v_proj.weight": "model-00010-of-00011.safetensors",
278
+ "model.layers.36.input_layernorm.weight": "model-00010-of-00011.safetensors",
279
+ "model.layers.36.mlp.down_proj.weight": "model-00010-of-00011.safetensors",
280
+ "model.layers.36.mlp.gate_proj.weight": "model-00010-of-00011.safetensors",
281
+ "model.layers.36.mlp.up_proj.weight": "model-00010-of-00011.safetensors",
282
+ "model.layers.36.post_attention_layernorm.weight": "model-00010-of-00011.safetensors",
283
+ "model.layers.36.self_attn.k_proj.weight": "model-00010-of-00011.safetensors",
284
+ "model.layers.36.self_attn.o_proj.weight": "model-00010-of-00011.safetensors",
285
+ "model.layers.36.self_attn.q_proj.weight": "model-00010-of-00011.safetensors",
286
+ "model.layers.36.self_attn.v_proj.weight": "model-00010-of-00011.safetensors",
287
+ "model.layers.37.input_layernorm.weight": "model-00010-of-00011.safetensors",
288
+ "model.layers.37.mlp.down_proj.weight": "model-00010-of-00011.safetensors",
289
+ "model.layers.37.mlp.gate_proj.weight": "model-00010-of-00011.safetensors",
290
+ "model.layers.37.mlp.up_proj.weight": "model-00010-of-00011.safetensors",
291
+ "model.layers.37.post_attention_layernorm.weight": "model-00010-of-00011.safetensors",
292
+ "model.layers.37.self_attn.k_proj.weight": "model-00010-of-00011.safetensors",
293
+ "model.layers.37.self_attn.o_proj.weight": "model-00010-of-00011.safetensors",
294
+ "model.layers.37.self_attn.q_proj.weight": "model-00010-of-00011.safetensors",
295
+ "model.layers.37.self_attn.v_proj.weight": "model-00010-of-00011.safetensors",
296
+ "model.layers.38.input_layernorm.weight": "model-00011-of-00011.safetensors",
297
+ "model.layers.38.mlp.down_proj.weight": "model-00011-of-00011.safetensors",
298
+ "model.layers.38.mlp.gate_proj.weight": "model-00011-of-00011.safetensors",
299
+ "model.layers.38.mlp.up_proj.weight": "model-00011-of-00011.safetensors",
300
+ "model.layers.38.post_attention_layernorm.weight": "model-00011-of-00011.safetensors",
301
+ "model.layers.38.self_attn.k_proj.weight": "model-00010-of-00011.safetensors",
302
+ "model.layers.38.self_attn.o_proj.weight": "model-00011-of-00011.safetensors",
303
+ "model.layers.38.self_attn.q_proj.weight": "model-00010-of-00011.safetensors",
304
+ "model.layers.38.self_attn.v_proj.weight": "model-00011-of-00011.safetensors",
305
+ "model.layers.39.input_layernorm.weight": "model-00011-of-00011.safetensors",
306
+ "model.layers.39.mlp.down_proj.weight": "model-00011-of-00011.safetensors",
307
+ "model.layers.39.mlp.gate_proj.weight": "model-00011-of-00011.safetensors",
308
+ "model.layers.39.mlp.up_proj.weight": "model-00011-of-00011.safetensors",
309
+ "model.layers.39.post_attention_layernorm.weight": "model-00011-of-00011.safetensors",
310
+ "model.layers.39.self_attn.k_proj.weight": "model-00011-of-00011.safetensors",
311
+ "model.layers.39.self_attn.o_proj.weight": "model-00011-of-00011.safetensors",
312
+ "model.layers.39.self_attn.q_proj.weight": "model-00011-of-00011.safetensors",
313
+ "model.layers.39.self_attn.v_proj.weight": "model-00011-of-00011.safetensors",
314
+ "model.layers.4.input_layernorm.weight": "model-00002-of-00011.safetensors",
315
+ "model.layers.4.mlp.down_proj.weight": "model-00002-of-00011.safetensors",
316
+ "model.layers.4.mlp.gate_proj.weight": "model-00002-of-00011.safetensors",
317
+ "model.layers.4.mlp.up_proj.weight": "model-00002-of-00011.safetensors",
318
+ "model.layers.4.post_attention_layernorm.weight": "model-00002-of-00011.safetensors",
319
+ "model.layers.4.self_attn.k_proj.weight": "model-00002-of-00011.safetensors",
320
+ "model.layers.4.self_attn.o_proj.weight": "model-00002-of-00011.safetensors",
321
+ "model.layers.4.self_attn.q_proj.weight": "model-00002-of-00011.safetensors",
322
+ "model.layers.4.self_attn.v_proj.weight": "model-00002-of-00011.safetensors",
323
+ "model.layers.5.input_layernorm.weight": "model-00002-of-00011.safetensors",
324
+ "model.layers.5.mlp.down_proj.weight": "model-00002-of-00011.safetensors",
325
+ "model.layers.5.mlp.gate_proj.weight": "model-00002-of-00011.safetensors",
326
+ "model.layers.5.mlp.up_proj.weight": "model-00002-of-00011.safetensors",
327
+ "model.layers.5.post_attention_layernorm.weight": "model-00002-of-00011.safetensors",
328
+ "model.layers.5.self_attn.k_proj.weight": "model-00002-of-00011.safetensors",
329
+ "model.layers.5.self_attn.o_proj.weight": "model-00002-of-00011.safetensors",
330
+ "model.layers.5.self_attn.q_proj.weight": "model-00002-of-00011.safetensors",
331
+ "model.layers.5.self_attn.v_proj.weight": "model-00002-of-00011.safetensors",
332
+ "model.layers.6.input_layernorm.weight": "model-00002-of-00011.safetensors",
333
+ "model.layers.6.mlp.down_proj.weight": "model-00002-of-00011.safetensors",
334
+ "model.layers.6.mlp.gate_proj.weight": "model-00002-of-00011.safetensors",
335
+ "model.layers.6.mlp.up_proj.weight": "model-00002-of-00011.safetensors",
336
+ "model.layers.6.post_attention_layernorm.weight": "model-00002-of-00011.safetensors",
337
+ "model.layers.6.self_attn.k_proj.weight": "model-00002-of-00011.safetensors",
338
+ "model.layers.6.self_attn.o_proj.weight": "model-00002-of-00011.safetensors",
339
+ "model.layers.6.self_attn.q_proj.weight": "model-00002-of-00011.safetensors",
340
+ "model.layers.6.self_attn.v_proj.weight": "model-00002-of-00011.safetensors",
341
+ "model.layers.7.input_layernorm.weight": "model-00003-of-00011.safetensors",
342
+ "model.layers.7.mlp.down_proj.weight": "model-00003-of-00011.safetensors",
343
+ "model.layers.7.mlp.gate_proj.weight": "model-00003-of-00011.safetensors",
344
+ "model.layers.7.mlp.up_proj.weight": "model-00003-of-00011.safetensors",
345
+ "model.layers.7.post_attention_layernorm.weight": "model-00003-of-00011.safetensors",
346
+ "model.layers.7.self_attn.k_proj.weight": "model-00002-of-00011.safetensors",
347
+ "model.layers.7.self_attn.o_proj.weight": "model-00003-of-00011.safetensors",
348
+ "model.layers.7.self_attn.q_proj.weight": "model-00002-of-00011.safetensors",
349
+ "model.layers.7.self_attn.v_proj.weight": "model-00002-of-00011.safetensors",
350
+ "model.layers.8.input_layernorm.weight": "model-00003-of-00011.safetensors",
351
+ "model.layers.8.mlp.down_proj.weight": "model-00003-of-00011.safetensors",
352
+ "model.layers.8.mlp.gate_proj.weight": "model-00003-of-00011.safetensors",
353
+ "model.layers.8.mlp.up_proj.weight": "model-00003-of-00011.safetensors",
354
+ "model.layers.8.post_attention_layernorm.weight": "model-00003-of-00011.safetensors",
355
+ "model.layers.8.self_attn.k_proj.weight": "model-00003-of-00011.safetensors",
356
+ "model.layers.8.self_attn.o_proj.weight": "model-00003-of-00011.safetensors",
357
+ "model.layers.8.self_attn.q_proj.weight": "model-00003-of-00011.safetensors",
358
+ "model.layers.8.self_attn.v_proj.weight": "model-00003-of-00011.safetensors",
359
+ "model.layers.9.input_layernorm.weight": "model-00003-of-00011.safetensors",
360
+ "model.layers.9.mlp.down_proj.weight": "model-00003-of-00011.safetensors",
361
+ "model.layers.9.mlp.gate_proj.weight": "model-00003-of-00011.safetensors",
362
+ "model.layers.9.mlp.up_proj.weight": "model-00003-of-00011.safetensors",
363
+ "model.layers.9.post_attention_layernorm.weight": "model-00003-of-00011.safetensors",
364
+ "model.layers.9.self_attn.k_proj.weight": "model-00003-of-00011.safetensors",
365
+ "model.layers.9.self_attn.o_proj.weight": "model-00003-of-00011.safetensors",
366
+ "model.layers.9.self_attn.q_proj.weight": "model-00003-of-00011.safetensors",
367
+ "model.layers.9.self_attn.v_proj.weight": "model-00003-of-00011.safetensors",
368
+ "model.norm.weight": "model-00011-of-00011.safetensors"
369
+ }
370
+ }
ft_13b_NLSY79_with_birth_year_ckpt_3/special_tokens_map.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": true,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "</s>",
11
+ "lstrip": false,
12
+ "normalized": true,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "<unk>",
18
+ "lstrip": false,
19
+ "normalized": true,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "unk_token": {
24
+ "content": "<unk>",
25
+ "lstrip": false,
26
+ "normalized": true,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ }
30
+ }
ft_13b_NLSY79_with_birth_year_ckpt_3/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
ft_13b_NLSY79_with_birth_year_ckpt_3/tokenizer_config.json ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": true,
3
+ "add_eos_token": false,
4
+ "add_prefix_space": null,
5
+ "added_tokens_decoder": {
6
+ "0": {
7
+ "content": "<unk>",
8
+ "lstrip": false,
9
+ "normalized": true,
10
+ "rstrip": false,
11
+ "single_word": false,
12
+ "special": true
13
+ },
14
+ "1": {
15
+ "content": "<s>",
16
+ "lstrip": false,
17
+ "normalized": true,
18
+ "rstrip": false,
19
+ "single_word": false,
20
+ "special": true
21
+ },
22
+ "2": {
23
+ "content": "</s>",
24
+ "lstrip": false,
25
+ "normalized": true,
26
+ "rstrip": false,
27
+ "single_word": false,
28
+ "special": true
29
+ },
30
+ "32000": {
31
+ "content": "<pad>",
32
+ "lstrip": true,
33
+ "normalized": true,
34
+ "rstrip": true,
35
+ "single_word": false,
36
+ "special": false
37
+ }
38
+ },
39
+ "additional_special_tokens": [],
40
+ "bos_token": "<s>",
41
+ "chat_template": "{% if messages[0]['role'] == 'system' %}{% set loop_messages = messages[1:] %}{% set system_message = messages[0]['content'] %}{% else %}{% set loop_messages = messages %}{% set system_message = false %}{% endif %}{% for message in loop_messages %}{% if loop.index0 == 0 and system_message != false %}{% set content = '<<SYS>>\\n' + system_message + '\\n<</SYS>>\\n\\n' + message['content'] %}{% else %}{% set content = message['content'] %}{% endif %}{% if message['role'] == 'user' or message['role'] == 'tool' %}{{ bos_token + '[INST] ' + content + ' [/INST]' }}{% elif message['role'] == 'system' %}{{ '<<SYS>>\\n' + content + '\\n<</SYS>>\\n\\n' }}{% elif message['role'] == 'assistant' %}{{ ' ' + content + ' ' + eos_token }}{% endif %}{% endfor %}",
42
+ "clean_up_tokenization_spaces": false,
43
+ "eos_token": "</s>",
44
+ "legacy": false,
45
+ "model_max_length": 4096,
46
+ "pad_token": "<unk>",
47
+ "padding_side": "right",
48
+ "sp_model_kwargs": {},
49
+ "spaces_between_special_tokens": false,
50
+ "tokenizer_class": "LlamaTokenizer",
51
+ "unk_token": "<unk>",
52
+ "use_default_system_prompt": true
53
+ }
ft_13b_NLSY79_with_birth_year_ckpt_3/trainer_state.json ADDED
@@ -0,0 +1,687 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "eval_steps": 30,
6
+ "global_step": 90,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.03333333333333333,
13
+ "grad_norm": 6.756654739379883,
14
+ "learning_rate": 9.88888888888889e-06,
15
+ "loss": 0.9896,
16
+ "step": 1
17
+ },
18
+ {
19
+ "epoch": 0.06666666666666667,
20
+ "grad_norm": 4.388773441314697,
21
+ "learning_rate": 9.777777777777779e-06,
22
+ "loss": 0.6516,
23
+ "step": 2
24
+ },
25
+ {
26
+ "epoch": 0.1,
27
+ "grad_norm": 11.626699447631836,
28
+ "learning_rate": 9.666666666666667e-06,
29
+ "loss": 0.401,
30
+ "step": 3
31
+ },
32
+ {
33
+ "epoch": 0.13333333333333333,
34
+ "grad_norm": 3.3663454055786133,
35
+ "learning_rate": 9.555555555555556e-06,
36
+ "loss": 0.2648,
37
+ "step": 4
38
+ },
39
+ {
40
+ "epoch": 0.16666666666666666,
41
+ "grad_norm": 5.141721248626709,
42
+ "learning_rate": 9.444444444444445e-06,
43
+ "loss": 0.2652,
44
+ "step": 5
45
+ },
46
+ {
47
+ "epoch": 0.2,
48
+ "grad_norm": 0.9627482891082764,
49
+ "learning_rate": 9.333333333333334e-06,
50
+ "loss": 0.181,
51
+ "step": 6
52
+ },
53
+ {
54
+ "epoch": 0.23333333333333334,
55
+ "grad_norm": 0.7184298634529114,
56
+ "learning_rate": 9.222222222222224e-06,
57
+ "loss": 0.1733,
58
+ "step": 7
59
+ },
60
+ {
61
+ "epoch": 0.26666666666666666,
62
+ "grad_norm": 0.6345965266227722,
63
+ "learning_rate": 9.111111111111112e-06,
64
+ "loss": 0.171,
65
+ "step": 8
66
+ },
67
+ {
68
+ "epoch": 0.3,
69
+ "grad_norm": 0.4310895502567291,
70
+ "learning_rate": 9e-06,
71
+ "loss": 0.1578,
72
+ "step": 9
73
+ },
74
+ {
75
+ "epoch": 0.3333333333333333,
76
+ "grad_norm": 0.42963308095932007,
77
+ "learning_rate": 8.888888888888888e-06,
78
+ "loss": 0.1588,
79
+ "step": 10
80
+ },
81
+ {
82
+ "epoch": 0.36666666666666664,
83
+ "grad_norm": 0.3190704882144928,
84
+ "learning_rate": 8.777777777777778e-06,
85
+ "loss": 0.1528,
86
+ "step": 11
87
+ },
88
+ {
89
+ "epoch": 0.4,
90
+ "grad_norm": 0.34697574377059937,
91
+ "learning_rate": 8.666666666666668e-06,
92
+ "loss": 0.157,
93
+ "step": 12
94
+ },
95
+ {
96
+ "epoch": 0.43333333333333335,
97
+ "grad_norm": 0.2627966105937958,
98
+ "learning_rate": 8.555555555555556e-06,
99
+ "loss": 0.1548,
100
+ "step": 13
101
+ },
102
+ {
103
+ "epoch": 0.4666666666666667,
104
+ "grad_norm": 0.23730801045894623,
105
+ "learning_rate": 8.444444444444446e-06,
106
+ "loss": 0.1528,
107
+ "step": 14
108
+ },
109
+ {
110
+ "epoch": 0.5,
111
+ "grad_norm": 0.27259665727615356,
112
+ "learning_rate": 8.333333333333334e-06,
113
+ "loss": 0.1406,
114
+ "step": 15
115
+ },
116
+ {
117
+ "epoch": 0.5333333333333333,
118
+ "grad_norm": 0.19127672910690308,
119
+ "learning_rate": 8.222222222222222e-06,
120
+ "loss": 0.1443,
121
+ "step": 16
122
+ },
123
+ {
124
+ "epoch": 0.5666666666666667,
125
+ "grad_norm": 0.18087811768054962,
126
+ "learning_rate": 8.111111111111112e-06,
127
+ "loss": 0.1439,
128
+ "step": 17
129
+ },
130
+ {
131
+ "epoch": 0.6,
132
+ "grad_norm": 0.23132827877998352,
133
+ "learning_rate": 8.000000000000001e-06,
134
+ "loss": 0.1463,
135
+ "step": 18
136
+ },
137
+ {
138
+ "epoch": 0.6333333333333333,
139
+ "grad_norm": 0.17551013827323914,
140
+ "learning_rate": 7.88888888888889e-06,
141
+ "loss": 0.1373,
142
+ "step": 19
143
+ },
144
+ {
145
+ "epoch": 0.6666666666666666,
146
+ "grad_norm": 0.16079062223434448,
147
+ "learning_rate": 7.77777777777778e-06,
148
+ "loss": 0.1442,
149
+ "step": 20
150
+ },
151
+ {
152
+ "epoch": 0.7,
153
+ "grad_norm": 0.16452927887439728,
154
+ "learning_rate": 7.666666666666667e-06,
155
+ "loss": 0.1462,
156
+ "step": 21
157
+ },
158
+ {
159
+ "epoch": 0.7333333333333333,
160
+ "grad_norm": 0.144525945186615,
161
+ "learning_rate": 7.555555555555556e-06,
162
+ "loss": 0.1444,
163
+ "step": 22
164
+ },
165
+ {
166
+ "epoch": 0.7666666666666667,
167
+ "grad_norm": 0.14696849882602692,
168
+ "learning_rate": 7.444444444444445e-06,
169
+ "loss": 0.1387,
170
+ "step": 23
171
+ },
172
+ {
173
+ "epoch": 0.8,
174
+ "grad_norm": 0.16059017181396484,
175
+ "learning_rate": 7.333333333333333e-06,
176
+ "loss": 0.1315,
177
+ "step": 24
178
+ },
179
+ {
180
+ "epoch": 0.8333333333333334,
181
+ "grad_norm": 0.16093049943447113,
182
+ "learning_rate": 7.222222222222223e-06,
183
+ "loss": 0.138,
184
+ "step": 25
185
+ },
186
+ {
187
+ "epoch": 0.8666666666666667,
188
+ "grad_norm": 0.1766657531261444,
189
+ "learning_rate": 7.111111111111112e-06,
190
+ "loss": 0.1367,
191
+ "step": 26
192
+ },
193
+ {
194
+ "epoch": 0.9,
195
+ "grad_norm": 0.22861993312835693,
196
+ "learning_rate": 7e-06,
197
+ "loss": 0.1413,
198
+ "step": 27
199
+ },
200
+ {
201
+ "epoch": 0.9333333333333333,
202
+ "grad_norm": 0.1955043077468872,
203
+ "learning_rate": 6.88888888888889e-06,
204
+ "loss": 0.1472,
205
+ "step": 28
206
+ },
207
+ {
208
+ "epoch": 0.9666666666666667,
209
+ "grad_norm": 0.20372217893600464,
210
+ "learning_rate": 6.777777777777779e-06,
211
+ "loss": 0.143,
212
+ "step": 29
213
+ },
214
+ {
215
+ "epoch": 1.0,
216
+ "grad_norm": 0.2251114398241043,
217
+ "learning_rate": 6.666666666666667e-06,
218
+ "loss": 0.1437,
219
+ "step": 30
220
+ },
221
+ {
222
+ "epoch": 1.0,
223
+ "eval_loss": 0.1389642059803009,
224
+ "eval_runtime": 5.0801,
225
+ "eval_samples_per_second": 26.18,
226
+ "eval_steps_per_second": 0.984,
227
+ "step": 30
228
+ },
229
+ {
230
+ "epoch": 1.0333333333333334,
231
+ "grad_norm": 0.24110844731330872,
232
+ "learning_rate": 6.555555555555556e-06,
233
+ "loss": 0.1467,
234
+ "step": 31
235
+ },
236
+ {
237
+ "epoch": 1.0666666666666667,
238
+ "grad_norm": 0.13794952630996704,
239
+ "learning_rate": 6.444444444444445e-06,
240
+ "loss": 0.1328,
241
+ "step": 32
242
+ },
243
+ {
244
+ "epoch": 1.1,
245
+ "grad_norm": 0.1344776451587677,
246
+ "learning_rate": 6.333333333333333e-06,
247
+ "loss": 0.1416,
248
+ "step": 33
249
+ },
250
+ {
251
+ "epoch": 1.1333333333333333,
252
+ "grad_norm": 0.15418794751167297,
253
+ "learning_rate": 6.222222222222223e-06,
254
+ "loss": 0.1347,
255
+ "step": 34
256
+ },
257
+ {
258
+ "epoch": 1.1666666666666667,
259
+ "grad_norm": 0.12393444776535034,
260
+ "learning_rate": 6.111111111111112e-06,
261
+ "loss": 0.1387,
262
+ "step": 35
263
+ },
264
+ {
265
+ "epoch": 1.2,
266
+ "grad_norm": 0.12951317429542542,
267
+ "learning_rate": 6e-06,
268
+ "loss": 0.1279,
269
+ "step": 36
270
+ },
271
+ {
272
+ "epoch": 1.2333333333333334,
273
+ "grad_norm": 0.1566055864095688,
274
+ "learning_rate": 5.88888888888889e-06,
275
+ "loss": 0.1403,
276
+ "step": 37
277
+ },
278
+ {
279
+ "epoch": 1.2666666666666666,
280
+ "grad_norm": 0.11854217946529388,
281
+ "learning_rate": 5.777777777777778e-06,
282
+ "loss": 0.134,
283
+ "step": 38
284
+ },
285
+ {
286
+ "epoch": 1.3,
287
+ "grad_norm": 0.1250990331172943,
288
+ "learning_rate": 5.666666666666667e-06,
289
+ "loss": 0.1309,
290
+ "step": 39
291
+ },
292
+ {
293
+ "epoch": 1.3333333333333333,
294
+ "grad_norm": 0.12426821887493134,
295
+ "learning_rate": 5.555555555555557e-06,
296
+ "loss": 0.1348,
297
+ "step": 40
298
+ },
299
+ {
300
+ "epoch": 1.3666666666666667,
301
+ "grad_norm": 0.14265167713165283,
302
+ "learning_rate": 5.444444444444445e-06,
303
+ "loss": 0.1411,
304
+ "step": 41
305
+ },
306
+ {
307
+ "epoch": 1.4,
308
+ "grad_norm": 0.14097608625888824,
309
+ "learning_rate": 5.333333333333334e-06,
310
+ "loss": 0.1366,
311
+ "step": 42
312
+ },
313
+ {
314
+ "epoch": 1.4333333333333333,
315
+ "grad_norm": 0.13661041855812073,
316
+ "learning_rate": 5.2222222222222226e-06,
317
+ "loss": 0.1382,
318
+ "step": 43
319
+ },
320
+ {
321
+ "epoch": 1.4666666666666668,
322
+ "grad_norm": 0.15367251634597778,
323
+ "learning_rate": 5.1111111111111115e-06,
324
+ "loss": 0.1345,
325
+ "step": 44
326
+ },
327
+ {
328
+ "epoch": 1.5,
329
+ "grad_norm": 0.13000041246414185,
330
+ "learning_rate": 5e-06,
331
+ "loss": 0.1361,
332
+ "step": 45
333
+ },
334
+ {
335
+ "epoch": 1.5333333333333332,
336
+ "grad_norm": 0.1131439059972763,
337
+ "learning_rate": 4.888888888888889e-06,
338
+ "loss": 0.1339,
339
+ "step": 46
340
+ },
341
+ {
342
+ "epoch": 1.5666666666666667,
343
+ "grad_norm": 0.15624798834323883,
344
+ "learning_rate": 4.777777777777778e-06,
345
+ "loss": 0.135,
346
+ "step": 47
347
+ },
348
+ {
349
+ "epoch": 1.6,
350
+ "grad_norm": 0.13269546627998352,
351
+ "learning_rate": 4.666666666666667e-06,
352
+ "loss": 0.1334,
353
+ "step": 48
354
+ },
355
+ {
356
+ "epoch": 1.6333333333333333,
357
+ "grad_norm": 0.11152497678995132,
358
+ "learning_rate": 4.555555555555556e-06,
359
+ "loss": 0.1425,
360
+ "step": 49
361
+ },
362
+ {
363
+ "epoch": 1.6666666666666665,
364
+ "grad_norm": 0.13962796330451965,
365
+ "learning_rate": 4.444444444444444e-06,
366
+ "loss": 0.1342,
367
+ "step": 50
368
+ },
369
+ {
370
+ "epoch": 1.7,
371
+ "grad_norm": 0.12835395336151123,
372
+ "learning_rate": 4.333333333333334e-06,
373
+ "loss": 0.1352,
374
+ "step": 51
375
+ },
376
+ {
377
+ "epoch": 1.7333333333333334,
378
+ "grad_norm": 0.12098712474107742,
379
+ "learning_rate": 4.222222222222223e-06,
380
+ "loss": 0.1341,
381
+ "step": 52
382
+ },
383
+ {
384
+ "epoch": 1.7666666666666666,
385
+ "grad_norm": 0.11860781908035278,
386
+ "learning_rate": 4.111111111111111e-06,
387
+ "loss": 0.1347,
388
+ "step": 53
389
+ },
390
+ {
391
+ "epoch": 1.8,
392
+ "grad_norm": 0.11772364377975464,
393
+ "learning_rate": 4.000000000000001e-06,
394
+ "loss": 0.1342,
395
+ "step": 54
396
+ },
397
+ {
398
+ "epoch": 1.8333333333333335,
399
+ "grad_norm": 0.12149743735790253,
400
+ "learning_rate": 3.88888888888889e-06,
401
+ "loss": 0.1375,
402
+ "step": 55
403
+ },
404
+ {
405
+ "epoch": 1.8666666666666667,
406
+ "grad_norm": 0.13157767057418823,
407
+ "learning_rate": 3.777777777777778e-06,
408
+ "loss": 0.1354,
409
+ "step": 56
410
+ },
411
+ {
412
+ "epoch": 1.9,
413
+ "grad_norm": 0.11540118604898453,
414
+ "learning_rate": 3.6666666666666666e-06,
415
+ "loss": 0.1341,
416
+ "step": 57
417
+ },
418
+ {
419
+ "epoch": 1.9333333333333333,
420
+ "grad_norm": 0.14963074028491974,
421
+ "learning_rate": 3.555555555555556e-06,
422
+ "loss": 0.1359,
423
+ "step": 58
424
+ },
425
+ {
426
+ "epoch": 1.9666666666666668,
427
+ "grad_norm": 0.13901227712631226,
428
+ "learning_rate": 3.444444444444445e-06,
429
+ "loss": 0.1336,
430
+ "step": 59
431
+ },
432
+ {
433
+ "epoch": 2.0,
434
+ "grad_norm": 0.11229171603918076,
435
+ "learning_rate": 3.3333333333333333e-06,
436
+ "loss": 0.1369,
437
+ "step": 60
438
+ },
439
+ {
440
+ "epoch": 2.0,
441
+ "eval_loss": 0.13511531054973602,
442
+ "eval_runtime": 5.1098,
443
+ "eval_samples_per_second": 26.028,
444
+ "eval_steps_per_second": 0.979,
445
+ "step": 60
446
+ },
447
+ {
448
+ "epoch": 2.033333333333333,
449
+ "grad_norm": 0.12950368225574493,
450
+ "learning_rate": 3.2222222222222227e-06,
451
+ "loss": 0.1275,
452
+ "step": 61
453
+ },
454
+ {
455
+ "epoch": 2.066666666666667,
456
+ "grad_norm": 0.10755325108766556,
457
+ "learning_rate": 3.1111111111111116e-06,
458
+ "loss": 0.1384,
459
+ "step": 62
460
+ },
461
+ {
462
+ "epoch": 2.1,
463
+ "grad_norm": 0.12992867827415466,
464
+ "learning_rate": 3e-06,
465
+ "loss": 0.1364,
466
+ "step": 63
467
+ },
468
+ {
469
+ "epoch": 2.1333333333333333,
470
+ "grad_norm": 0.1376504898071289,
471
+ "learning_rate": 2.888888888888889e-06,
472
+ "loss": 0.1406,
473
+ "step": 64
474
+ },
475
+ {
476
+ "epoch": 2.1666666666666665,
477
+ "grad_norm": 0.11081744730472565,
478
+ "learning_rate": 2.7777777777777783e-06,
479
+ "loss": 0.1333,
480
+ "step": 65
481
+ },
482
+ {
483
+ "epoch": 2.2,
484
+ "grad_norm": 0.09679851680994034,
485
+ "learning_rate": 2.666666666666667e-06,
486
+ "loss": 0.136,
487
+ "step": 66
488
+ },
489
+ {
490
+ "epoch": 2.2333333333333334,
491
+ "grad_norm": 0.13199330866336823,
492
+ "learning_rate": 2.5555555555555557e-06,
493
+ "loss": 0.1294,
494
+ "step": 67
495
+ },
496
+ {
497
+ "epoch": 2.2666666666666666,
498
+ "grad_norm": 0.12924429774284363,
499
+ "learning_rate": 2.4444444444444447e-06,
500
+ "loss": 0.1349,
501
+ "step": 68
502
+ },
503
+ {
504
+ "epoch": 2.3,
505
+ "grad_norm": 0.1062309667468071,
506
+ "learning_rate": 2.3333333333333336e-06,
507
+ "loss": 0.133,
508
+ "step": 69
509
+ },
510
+ {
511
+ "epoch": 2.3333333333333335,
512
+ "grad_norm": 0.10067743808031082,
513
+ "learning_rate": 2.222222222222222e-06,
514
+ "loss": 0.1382,
515
+ "step": 70
516
+ },
517
+ {
518
+ "epoch": 2.3666666666666667,
519
+ "grad_norm": 0.14575141668319702,
520
+ "learning_rate": 2.1111111111111114e-06,
521
+ "loss": 0.1338,
522
+ "step": 71
523
+ },
524
+ {
525
+ "epoch": 2.4,
526
+ "grad_norm": 0.10778827965259552,
527
+ "learning_rate": 2.0000000000000003e-06,
528
+ "loss": 0.1363,
529
+ "step": 72
530
+ },
531
+ {
532
+ "epoch": 2.4333333333333336,
533
+ "grad_norm": 0.10342462360858917,
534
+ "learning_rate": 1.888888888888889e-06,
535
+ "loss": 0.1282,
536
+ "step": 73
537
+ },
538
+ {
539
+ "epoch": 2.466666666666667,
540
+ "grad_norm": 0.0964592695236206,
541
+ "learning_rate": 1.777777777777778e-06,
542
+ "loss": 0.1334,
543
+ "step": 74
544
+ },
545
+ {
546
+ "epoch": 2.5,
547
+ "grad_norm": 0.12873083353042603,
548
+ "learning_rate": 1.6666666666666667e-06,
549
+ "loss": 0.1373,
550
+ "step": 75
551
+ },
552
+ {
553
+ "epoch": 2.533333333333333,
554
+ "grad_norm": 0.14387091994285583,
555
+ "learning_rate": 1.5555555555555558e-06,
556
+ "loss": 0.1354,
557
+ "step": 76
558
+ },
559
+ {
560
+ "epoch": 2.5666666666666664,
561
+ "grad_norm": 0.12266004085540771,
562
+ "learning_rate": 1.4444444444444445e-06,
563
+ "loss": 0.1309,
564
+ "step": 77
565
+ },
566
+ {
567
+ "epoch": 2.6,
568
+ "grad_norm": 0.08805371820926666,
569
+ "learning_rate": 1.3333333333333334e-06,
570
+ "loss": 0.1341,
571
+ "step": 78
572
+ },
573
+ {
574
+ "epoch": 2.6333333333333333,
575
+ "grad_norm": 0.11398159712553024,
576
+ "learning_rate": 1.2222222222222223e-06,
577
+ "loss": 0.1336,
578
+ "step": 79
579
+ },
580
+ {
581
+ "epoch": 2.6666666666666665,
582
+ "grad_norm": 0.10753536969423294,
583
+ "learning_rate": 1.111111111111111e-06,
584
+ "loss": 0.1304,
585
+ "step": 80
586
+ },
587
+ {
588
+ "epoch": 2.7,
589
+ "grad_norm": 0.12526370584964752,
590
+ "learning_rate": 1.0000000000000002e-06,
591
+ "loss": 0.1359,
592
+ "step": 81
593
+ },
594
+ {
595
+ "epoch": 2.7333333333333334,
596
+ "grad_norm": 0.1131182312965393,
597
+ "learning_rate": 8.88888888888889e-07,
598
+ "loss": 0.1306,
599
+ "step": 82
600
+ },
601
+ {
602
+ "epoch": 2.7666666666666666,
603
+ "grad_norm": 0.10941901803016663,
604
+ "learning_rate": 7.777777777777779e-07,
605
+ "loss": 0.1271,
606
+ "step": 83
607
+ },
608
+ {
609
+ "epoch": 2.8,
610
+ "grad_norm": 0.09163400530815125,
611
+ "learning_rate": 6.666666666666667e-07,
612
+ "loss": 0.1346,
613
+ "step": 84
614
+ },
615
+ {
616
+ "epoch": 2.8333333333333335,
617
+ "grad_norm": 0.09697934240102768,
618
+ "learning_rate": 5.555555555555555e-07,
619
+ "loss": 0.1265,
620
+ "step": 85
621
+ },
622
+ {
623
+ "epoch": 2.8666666666666667,
624
+ "grad_norm": 0.098645880818367,
625
+ "learning_rate": 4.444444444444445e-07,
626
+ "loss": 0.1284,
627
+ "step": 86
628
+ },
629
+ {
630
+ "epoch": 2.9,
631
+ "grad_norm": 0.11068296432495117,
632
+ "learning_rate": 3.3333333333333335e-07,
633
+ "loss": 0.1318,
634
+ "step": 87
635
+ },
636
+ {
637
+ "epoch": 2.9333333333333336,
638
+ "grad_norm": 0.10501579940319061,
639
+ "learning_rate": 2.2222222222222224e-07,
640
+ "loss": 0.1256,
641
+ "step": 88
642
+ },
643
+ {
644
+ "epoch": 2.966666666666667,
645
+ "grad_norm": 0.11091522872447968,
646
+ "learning_rate": 1.1111111111111112e-07,
647
+ "loss": 0.1328,
648
+ "step": 89
649
+ },
650
+ {
651
+ "epoch": 3.0,
652
+ "grad_norm": 0.10818880051374435,
653
+ "learning_rate": 0.0,
654
+ "loss": 0.1254,
655
+ "step": 90
656
+ },
657
+ {
658
+ "epoch": 3.0,
659
+ "eval_loss": 0.13397298753261566,
660
+ "eval_runtime": 5.2044,
661
+ "eval_samples_per_second": 25.555,
662
+ "eval_steps_per_second": 0.961,
663
+ "step": 90
664
+ }
665
+ ],
666
+ "logging_steps": 1.0,
667
+ "max_steps": 90,
668
+ "num_input_tokens_seen": 0,
669
+ "num_train_epochs": 3,
670
+ "save_steps": 30,
671
+ "stateful_callbacks": {
672
+ "TrainerControl": {
673
+ "args": {
674
+ "should_epoch_stop": false,
675
+ "should_evaluate": false,
676
+ "should_log": false,
677
+ "should_save": true,
678
+ "should_training_stop": true
679
+ },
680
+ "attributes": {}
681
+ }
682
+ },
683
+ "total_flos": 1.133858697332654e+17,
684
+ "train_batch_size": 4,
685
+ "trial_name": null,
686
+ "trial_params": null
687
+ }