HarleyCooper commited on
Commit
08c7807
·
verified ·
1 Parent(s): 15353a0

Upload folder using huggingface_hub

Browse files
Files changed (5) hide show
  1. README.md +304 -0
  2. config.json +10 -0
  3. model_001560.pt +3 -0
  4. token_bytes.pt +3 -0
  5. tokenizer.pkl +3 -0
README.md ADDED
@@ -0,0 +1,304 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # nanochat training report
2
+
3
+ Generated: 2025-10-23 22:26:58
4
+
5
+ ## Environment
6
+
7
+ ### Git Information
8
+ - Branch: main
9
+ - Commit: f1e8954 (dirty)
10
+ - Message: Document Hyperbolic setup and fix lite RL script
11
+
12
+ ### Hardware
13
+ - Platform: Linux
14
+ - CPUs: 104 cores (104 logical)
15
+ - Memory: 1007.4 GB
16
+ - GPUs: 8x NVIDIA H100 80GB HBM3
17
+ - GPU Memory: 633.7 GB total
18
+ - CUDA Version: 12.8
19
+ - Hourly Rate: $24.00/hour
20
+
21
+ ### Software
22
+ - Python: 3.10.12
23
+ - PyTorch: 2.8.0+cu128
24
+
25
+
26
+ ### Bloat
27
+ - Characters: 474,203
28
+ - Lines: 12,350
29
+ - Files: 57
30
+ - Tokens (approx): 118,550
31
+ - Dependencies (uv.lock lines): 2,220
32
+
33
+ Run started: 2025-10-23 22:27:01
34
+
35
+ ---
36
+
37
+ ## Tokenizer training
38
+ timestamp: 2025-10-24 01:20:23
39
+
40
+ - max_chars: 10,000,000,000
41
+ - doc_cap: 10,000
42
+ - vocab_size: 65,536
43
+ - train_time: 167.3429
44
+ - num_special_tokens: 9
45
+ - token_bytes_min: 1
46
+ - token_bytes_max: 32
47
+ - token_bytes_mean: 6.9125
48
+ - token_bytes_std: 2.8738
49
+
50
+
51
+ ## Tokenizer evaluation
52
+ timestamp: 2025-10-24 01:20:28
53
+
54
+ ### Comparison with GPT-2
55
+
56
+ | Text Type | Bytes | GPT-2 Tokens | GPT-2 Ratio | Ours Tokens | Ours Ratio | Relative Diff % |
57
+ |-----------|-------|--------------|--------------|-------------|------------|-----------------|
58
+ | news | 1819 | 404 | 4.50 | 371 | 4.90 | +8.2% |
59
+ | korean | 893 | 745 | 1.20 | 723 | 1.24 | +3.0% |
60
+ | code | 1259 | 576 | 2.19 | 492 | 2.56 | +14.6% |
61
+ | math | 1834 | 936 | 1.96 | 966 | 1.90 | -3.2% |
62
+ | science | 1112 | 260 | 4.28 | 223 | 4.99 | +14.2% |
63
+ | fwe-train | 4208518 | 900364 | 4.67 | 856938 | 4.91 | +4.8% |
64
+ | fwe-val | 5028883 | 1083776 | 4.64 | 1033017 | 4.87 | +4.7% |
65
+
66
+ ### Comparison with GPT-4
67
+
68
+ | Text Type | Bytes | GPT-4 Tokens | GPT-4 Ratio | Ours Tokens | Ours Ratio | Relative Diff % |
69
+ |-----------|-------|--------------|--------------|-------------|------------|-----------------|
70
+ | news | 1819 | 387 | 4.70 | 371 | 4.90 | +4.1% |
71
+ | korean | 893 | 364 | 2.45 | 723 | 1.24 | -98.6% |
72
+ | code | 1259 | 309 | 4.07 | 492 | 2.56 | -59.2% |
73
+ | math | 1834 | 832 | 2.20 | 966 | 1.90 | -16.1% |
74
+ | science | 1112 | 249 | 4.47 | 223 | 4.99 | +10.4% |
75
+ | fwe-train | 4208518 | 874799 | 4.81 | 856938 | 4.91 | +2.0% |
76
+ | fwe-val | 5028883 | 1054265 | 4.77 | 1033017 | 4.87 | +2.0% |
77
+
78
+
79
+ ## Base model training
80
+ timestamp: 2025-10-23 22:28:41
81
+
82
+ - run: aquarat-20251023-222655
83
+ - device_type:
84
+ - depth: 8
85
+ - max_seq_len: 2048
86
+ - num_iterations: 200
87
+ - target_flops: -1.0000
88
+ - target_param_data_ratio: 20
89
+ - device_batch_size: 32
90
+ - total_batch_size: 524,288
91
+ - embedding_lr: 0.2000
92
+ - unembedding_lr: 0.0040
93
+ - weight_decay: 0.0000
94
+ - matrix_lr: 0.0200
95
+ - grad_clip: 1.0000
96
+ - eval_every: 250
97
+ - eval_tokens: 10,485,760
98
+ - core_metric_every: 2000
99
+ - core_metric_max_per_task: 500
100
+ - sample_every: 2000
101
+ - model_tag:
102
+ - Number of parameters: 92,274,688
103
+ - Number of FLOPs per token: 4.529848e+08
104
+ - Calculated number of iterations: 200
105
+ - Number of training tokens: 104,857,600
106
+ - Tokens : Params ratio: 1.1364
107
+ - DDP world size: 8
108
+ - warmup_ratio: 0.0000
109
+ - warmdown_ratio: 0.2000
110
+ - final_lr_frac: 0.0000
111
+ - Minimum validation bpb: 1.2992
112
+ - Final validation bpb: 1.2992
113
+ - CORE metric estimate: 0.0135
114
+ - MFU %: 21.07%
115
+ - Total training flops: 4.749890e+16
116
+ - Total training time: 0.39m
117
+ - Peak memory usage: 19176.26MiB
118
+
119
+
120
+ ## Base model loss
121
+ timestamp: 2025-10-24 01:21:08
122
+
123
+ - train bpb: 1.2967
124
+ - val bpb: 1.3001
125
+ - sample 0: <|bos|>The capital of France is the capital of the city of the city of the city of the city of the
126
+ - sample 1: <|bos|>The chemical symbol of gold is a symbol of the world’s most important symbol of the world’s most important symbol
127
+ - sample 2: <|bos|>If yesterday was Friday, then tomorrow will be a bit more than a few years, and then the next day, the next
128
+ - sample 3: <|bos|>The opposite of hot is the same as the one of the two of the same kind of cold weather.
129
+ - sample 4: <|bos|>The planets of the solar system are: 1.5 billion years ago, and the 1.5 billion years
130
+ - sample 5: <|bos|>My favorite color is the color of the color of the color of the color of the color of the
131
+ - sample 6: <|bos|>If 5*x + 3 = 13, then x is 3.5.5.5.5.5.5.5
132
+
133
+
134
+ ## Base model evaluation
135
+ timestamp: 2025-10-24 01:32:41
136
+
137
+ - Model: base_model (step 200)
138
+ - CORE metric: 0.0155
139
+ - hellaswag_zeroshot: -0.0018
140
+ - jeopardy: 0.0000
141
+ - bigbench_qa_wikidata: 0.0005
142
+ - arc_easy: 0.1246
143
+ - arc_challenge: -0.0364
144
+ - copa: -0.0400
145
+ - commonsense_qa: -0.0053
146
+ - piqa: 0.0794
147
+ - openbook_qa: -0.0373
148
+ - lambada_openai: 0.0190
149
+ - hellaswag: -0.0067
150
+ - winograd: 0.0842
151
+ - winogrande: 0.0024
152
+ - bigbench_dyck_languages: 0.0080
153
+ - agi_eval_lsat_ar: 0.0326
154
+ - bigbench_cs_algorithms: 0.0235
155
+ - bigbench_operators: 0.0619
156
+ - bigbench_repeat_copy_logic: 0.0000
157
+ - squad: 0.0004
158
+ - coqa: 0.0009
159
+ - boolq: -0.1468
160
+ - bigbench_language_identification: 0.1770
161
+
162
+
163
+ ## Midtraining
164
+ timestamp: 2025-10-23 22:30:54
165
+
166
+ - run: aquarat-20251023-222655
167
+ - device_type:
168
+ - dtype: bfloat16
169
+ - num_iterations: 200
170
+ - max_seq_len: 2048
171
+ - device_batch_size: 32
172
+ - unembedding_lr: 0.0040
173
+ - embedding_lr: 0.2000
174
+ - matrix_lr: 0.0200
175
+ - init_lr_frac: 1.0000
176
+ - weight_decay: 0.0000
177
+ - eval_every: 150
178
+ - eval_tokens: 10,485,760
179
+ - total_batch_size: 524,288
180
+ - dry_run: 0
181
+ - Number of iterations: 199
182
+ - DDP world size: 8
183
+ - Minimum validation bpb: 0.6738
184
+
185
+
186
+ ## Chat evaluation mid
187
+ timestamp: 2025-10-24 01:52:32
188
+
189
+ - source: mid
190
+ - task_name: GSM8K
191
+ - dtype: bfloat16
192
+ - temperature: 0.0000
193
+ - max_new_tokens: 512
194
+ - num_samples: 1
195
+ - top_k: 50
196
+ - batch_size: 8
197
+ - model_tag: None
198
+ - step: None
199
+ - max_problems: None
200
+ - device_type:
201
+ - GSM8K: 0.0023
202
+
203
+
204
+ ## Chat SFT
205
+ timestamp: 2025-10-23 22:46:36
206
+
207
+ - run: aquarat-20251023-224449
208
+ - source: mid
209
+ - device_type:
210
+ - dtype: bfloat16
211
+ - device_batch_size: 4
212
+ - num_epochs: 1
213
+ - num_iterations: -1
214
+ - target_examples_per_step: 32
215
+ - unembedding_lr: 0.0040
216
+ - embedding_lr: 0.2000
217
+ - matrix_lr: 0.0200
218
+ - weight_decay: 0.0000
219
+ - init_lr_frac: 0.0200
220
+ - aqua_train_examples: 20,000
221
+ - aqua_val_examples: 254
222
+ - eval_every: 100
223
+ - eval_steps: 100
224
+ - eval_metrics_every: 200
225
+ - eval_metrics_max_problems: 1024
226
+ - Training rows: 41,839
227
+ - Number of iterations: 1307
228
+ - Training loss: 2.9144
229
+ - Validation loss: 1.9120
230
+
231
+
232
+ ## Chat evaluation sft
233
+ timestamp: 2025-10-24 01:52:40
234
+
235
+ - source: sft
236
+ - task_name: AQUA
237
+ - dtype: bfloat16
238
+ - temperature: 0.0000
239
+ - max_new_tokens: 512
240
+ - num_samples: 1
241
+ - top_k: 50
242
+ - batch_size: 8
243
+ - model_tag: None
244
+ - step: None
245
+ - max_problems: None
246
+ - device_type:
247
+ - AQUA: 0.2756
248
+
249
+
250
+ ## Chat RL
251
+ timestamp: 2025-10-24 01:01:09
252
+
253
+ - run: aquarat-20251023-224750-rl
254
+ - source: sft
255
+ - dtype: bfloat16
256
+ - device_batch_size: 1
257
+ - examples_per_step: 16
258
+ - num_samples: 4
259
+ - max_new_tokens: 64
260
+ - temperature: 0.7000
261
+ - top_k: 50
262
+ - unembedding_lr: 0.0040
263
+ - embedding_lr: 0.2000
264
+ - matrix_lr: 0.0200
265
+ - weight_decay: 0.0000
266
+ - init_lr_frac: 0.0500
267
+ - num_epochs: 1
268
+ - save_every: 60
269
+ - eval_every: 60
270
+ - eval_examples: 400
271
+
272
+
273
+ ## Chat evaluation rl
274
+ timestamp: 2025-10-24 04:11:26
275
+
276
+ - source: rl
277
+ - task_name: AQUA
278
+ - dtype: bfloat16
279
+ - temperature: 0.0000
280
+ - max_new_tokens: 64
281
+ - num_samples: 1
282
+ - top_k: 50
283
+ - batch_size: 8
284
+ - model_tag: None
285
+ - step: None
286
+ - max_problems: None
287
+ - device_type:
288
+ - AQUA: 0.2717
289
+
290
+
291
+ ## Summary
292
+
293
+ - Characters: 474,203
294
+ - Lines: 12,350
295
+ - Files: 57
296
+ - Tokens (approx): 118,550
297
+ - Dependencies (uv.lock lines): 2,220
298
+
299
+ | Metric | BASE | MID | SFT | RL |
300
+ |-----------------|----------|----------|----------|----------|
301
+ | CORE | 0.0155 | - | - | - |
302
+ | GSM8K | - | 0.0023 | - | - |
303
+
304
+ Total wall clock time: 3h25m
config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "base_checkpoint": "model_001560.pt",
3
+ "tokenizer": "tokenizer.pkl",
4
+ "train_dataset": "deepmind/aqua_rat",
5
+ "rl_steps": 1560,
6
+ "accuracy": {
7
+ "rl_aqua_dev": 0.2717,
8
+ "sft_aqua_dev": 0.2756
9
+ }
10
+ }
model_001560.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:249fb9f5b32a9e2110bdb9250ef09d2bd4927ec718a301e4432a01b04423cf0d
3
+ size 302010597
token_bytes.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f02a63722018ce8eb640f6fa455c4f5f6b3e2d1d4cf8842db47e96f57eb9e21
3
+ size 263721
tokenizer.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:762048e638ab63a4445d4d1bd404f8e9b317136ebe523b2f7cb6fe0f1b00be01
3
+ size 846033