File size: 17,339 Bytes
7b5acd6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 | {
"checkpoint_config": {
"activation": "gelu",
"actual_precision": "bf16",
"adamw_betas": [
0.9,
0.95
],
"adamw_eps": 1e-08,
"architecture": "gpt2",
"attention_kernel_policy": "auto",
"batch_size": 2,
"benchmark": {
"enable_central_tensorboard": true,
"enable_local_tensorboard": true,
"enabled": false,
"output_path": "/mnt/apps/llm-nanochat/artifacts/runs/20260715_resume-gpt2medium-gpt2preln-k20-optimizeronly-cpt14700-step34000-d1700-webwiki/throughput_benchmark.json",
"warmup_steps": 0
},
"bias": true,
"block_type": "gpt2_prelayernorm",
"causal_mask_mode": "buffered_upper_triangular",
"checkpoint_dir": "/mnt/apps/llm-nanochat/checkpoints/20260715_resume-gpt2medium-gpt2preln-k20-optimizeronly-cpt14700-step34000-d1700-webwiki",
"clip_grad_norm": 1.0,
"compile": {
"backend": null,
"compile_setup_sec": 0.0,
"diagnostic": null,
"dynamic": false,
"enabled": false,
"error_policy": "raise",
"fullgraph": false,
"mode": null,
"requested": false,
"status": "disabled"
},
"dataset": {
"storage_mode": "indexed_jsonl"
},
"dataset_dir": "/mnt/apps/llm-nanochat/datasets/202605141153_fineweb50_wiki50_50en_50it_score100_2500context_5Btokens_tok_20260515_en50it50_webwiki_stratified_500M",
"decay_shape": "inverse_proportional",
"decay_steps": 1700,
"decay_steps_source": "explicit",
"deterministic_algorithms": false,
"device": "cuda",
"dim": 1024,
"dropout": 0.0,
"effective_global_step": 34200,
"effective_layer_count": 24,
"effective_learning_rate": 3.309994670030697e-05,
"final_lr": 1e-05,
"final_lr_source": "explicit",
"fp8_backend": null,
"grad_accum_steps": 48,
"init_from": "/mnt/apps/llm-nanochat/checkpoints/20260703_continual-pretraining-gpt2medium-gpt2preln-k20-step14700-lr5e5-w500-s18500-d2000-final1e5-webwiki/step_34000.pt",
"init_strategy": "gpt2_std_0.02_residual_scale",
"learning_rate": 4.7832243797285384e-05,
"local_max_steps": 35700,
"local_step": 200,
"logging": {
"enable_central_tensorboard": true,
"enable_local_tensorboard": true,
"metrics_flush_every_steps": 1,
"metrics_writer": "persistent_jsonl_handle"
},
"lr": 4.7832243797285384e-05,
"lr_schedule": "wsd-decay-only",
"max_seq_len": 2500,
"max_steps": 35700,
"n_heads": 16,
"n_layers": 24,
"norm_order": "preln",
"norm_type": "layernorm",
"optimizer": {
"backend": "torch",
"betas": [
0.9,
0.95
],
"eps": 1e-08,
"implementation": "torch.optim.AdamW",
"learning_rate": 4.7832243797285384e-05,
"state_precision": "full_precision",
"type": "adamw",
"weight_decay": 0.1
},
"optimizer_backend": "torch",
"optimizer_implementation": "torch.optim.AdamW",
"optimizer_reset": false,
"optimizer_state_precision": "full_precision",
"optimizer_type": "adamw",
"peak_lr": 4.7832243797285384e-05,
"positional_encoding": "learned_absolute",
"repro": {
"attention_kernel_policy": "auto",
"cublas_workspace_config": null,
"cudnn_benchmark": true,
"cudnn_deterministic": false,
"deterministic_algorithms": false,
"flash_sdp_enabled": true,
"math_sdp_enabled": true,
"mem_efficient_sdp_enabled": true,
"pythonhashseed": "1337",
"seed": 1337
},
"requested_precision": "bf16",
"resume_checkpoint_step": 34000,
"resume_from": "/mnt/apps/llm-nanochat/checkpoints/20260703_continual-pretraining-gpt2medium-gpt2preln-k20-step14700-lr5e5-w500-s18500-d2000-final1e5-webwiki/step_34000.pt",
"resume_mode": "optimizer_only",
"resume_step": 34000,
"save_every_steps": 100,
"save_final_checkpoint": true,
"schedule_total_steps": 35700,
"scheduler": {
"decay_shape": "inverse_proportional",
"decay_steps": 1700,
"decay_steps_source": "explicit",
"final_lr": 1e-05,
"final_lr_source": "explicit",
"peak_lr": 4.7832243797285384e-05,
"resume_step": 34000,
"schedule_type": "wsd-decay-only",
"stable_steps": 0,
"target_final_step": 35700,
"total_steps": 35700,
"warmup_steps": 0
},
"scheduler_reset": true,
"seed": 1337,
"source_global_step": 34000,
"stable_steps": 0,
"target_final_step": 35700,
"target_global_step": 35700,
"tie_word_embeddings": true,
"train_cache_ram_bytes": 1073741824,
"train_cache_ram_mb": 1024,
"vocab_size": 32000,
"warmup_steps": 0,
"weight_decay": 0.1
},
"checkpoint_path": "/mnt/apps/llm-nanochat/checkpoints/20260715_resume-gpt2medium-gpt2preln-k20-optimizeronly-cpt14700-step34000-d1700-webwiki/step_34200.pt",
"config_path": "/mnt/apps/llm-nanochat/hf_exports/1gpu-llm-medium-v2/config.json",
"exported_at": "2026-07-16T13:41:05.693620+00:00",
"format": "llm-nanochat-safetensors-export",
"global_step": 34200,
"metadata_path": "/mnt/apps/llm-nanochat/hf_exports/1gpu-llm-medium-v2/step_34200.safetensors.json",
"model_config": {
"activation": "gelu",
"architecture": "gpt2",
"bias": true,
"block_type": "gpt2_prelayernorm",
"causal_mask_mode": "buffered_upper_triangular",
"dim": 1024,
"dropout": 0.0,
"init_strategy": "gpt2_std_0.02_residual_scale",
"max_seq_len": 2500,
"n_heads": 16,
"n_layers": 24,
"norm_order": "preln",
"norm_type": "layernorm",
"positional_encoding": "learned_absolute",
"tie_word_embeddings": true,
"vocab_size": 32000
},
"num_parameters": 370439424,
"num_tensors": 294,
"provenance": {
"checkpoint_dir": "/mnt/apps/llm-nanochat/checkpoints/20260715_resume-gpt2medium-gpt2preln-k20-optimizeronly-cpt14700-step34000-d1700-webwiki",
"checkpoint_name": "step_34200.pt",
"checkpoint_path": "/mnt/apps/llm-nanochat/checkpoints/20260715_resume-gpt2medium-gpt2preln-k20-optimizeronly-cpt14700-step34000-d1700-webwiki/step_34200.pt",
"global_step": 34200,
"packed_dataset_config_path": null,
"run_dir": "/mnt/apps/llm-nanochat/checkpoints",
"tokenizer_dir": "/mnt/apps/llm-nanochat/tokenizers/tokenizer_20260515_en50it50_webwiki_stratified_500M",
"training_config_path": null
},
"safetensors_path": "/mnt/apps/llm-nanochat/hf_exports/1gpu-llm-medium-v2/step_34200.safetensors",
"source_checkpoint_path": "/mnt/apps/llm-nanochat/checkpoints/20260715_resume-gpt2medium-gpt2preln-k20-optimizeronly-cpt14700-step34000-d1700-webwiki/step_34200.pt",
"source_global_step": 34200,
"tensor_names": [
"token_emb.weight",
"pos_emb.weight",
"blocks.0.ln_1.weight",
"blocks.0.ln_1.bias",
"blocks.0.ln_2.weight",
"blocks.0.ln_2.bias",
"blocks.0.attn.qkv.weight",
"blocks.0.attn.qkv.bias",
"blocks.0.attn.out_proj.weight",
"blocks.0.attn.out_proj.bias",
"blocks.0.mlp.fc.weight",
"blocks.0.mlp.fc.bias",
"blocks.0.mlp.proj.weight",
"blocks.0.mlp.proj.bias",
"blocks.1.ln_1.weight",
"blocks.1.ln_1.bias",
"blocks.1.ln_2.weight",
"blocks.1.ln_2.bias",
"blocks.1.attn.qkv.weight",
"blocks.1.attn.qkv.bias",
"blocks.1.attn.out_proj.weight",
"blocks.1.attn.out_proj.bias",
"blocks.1.mlp.fc.weight",
"blocks.1.mlp.fc.bias",
"blocks.1.mlp.proj.weight",
"blocks.1.mlp.proj.bias",
"blocks.2.ln_1.weight",
"blocks.2.ln_1.bias",
"blocks.2.ln_2.weight",
"blocks.2.ln_2.bias",
"blocks.2.attn.qkv.weight",
"blocks.2.attn.qkv.bias",
"blocks.2.attn.out_proj.weight",
"blocks.2.attn.out_proj.bias",
"blocks.2.mlp.fc.weight",
"blocks.2.mlp.fc.bias",
"blocks.2.mlp.proj.weight",
"blocks.2.mlp.proj.bias",
"blocks.3.ln_1.weight",
"blocks.3.ln_1.bias",
"blocks.3.ln_2.weight",
"blocks.3.ln_2.bias",
"blocks.3.attn.qkv.weight",
"blocks.3.attn.qkv.bias",
"blocks.3.attn.out_proj.weight",
"blocks.3.attn.out_proj.bias",
"blocks.3.mlp.fc.weight",
"blocks.3.mlp.fc.bias",
"blocks.3.mlp.proj.weight",
"blocks.3.mlp.proj.bias",
"blocks.4.ln_1.weight",
"blocks.4.ln_1.bias",
"blocks.4.ln_2.weight",
"blocks.4.ln_2.bias",
"blocks.4.attn.qkv.weight",
"blocks.4.attn.qkv.bias",
"blocks.4.attn.out_proj.weight",
"blocks.4.attn.out_proj.bias",
"blocks.4.mlp.fc.weight",
"blocks.4.mlp.fc.bias",
"blocks.4.mlp.proj.weight",
"blocks.4.mlp.proj.bias",
"blocks.5.ln_1.weight",
"blocks.5.ln_1.bias",
"blocks.5.ln_2.weight",
"blocks.5.ln_2.bias",
"blocks.5.attn.qkv.weight",
"blocks.5.attn.qkv.bias",
"blocks.5.attn.out_proj.weight",
"blocks.5.attn.out_proj.bias",
"blocks.5.mlp.fc.weight",
"blocks.5.mlp.fc.bias",
"blocks.5.mlp.proj.weight",
"blocks.5.mlp.proj.bias",
"blocks.6.ln_1.weight",
"blocks.6.ln_1.bias",
"blocks.6.ln_2.weight",
"blocks.6.ln_2.bias",
"blocks.6.attn.qkv.weight",
"blocks.6.attn.qkv.bias",
"blocks.6.attn.out_proj.weight",
"blocks.6.attn.out_proj.bias",
"blocks.6.mlp.fc.weight",
"blocks.6.mlp.fc.bias",
"blocks.6.mlp.proj.weight",
"blocks.6.mlp.proj.bias",
"blocks.7.ln_1.weight",
"blocks.7.ln_1.bias",
"blocks.7.ln_2.weight",
"blocks.7.ln_2.bias",
"blocks.7.attn.qkv.weight",
"blocks.7.attn.qkv.bias",
"blocks.7.attn.out_proj.weight",
"blocks.7.attn.out_proj.bias",
"blocks.7.mlp.fc.weight",
"blocks.7.mlp.fc.bias",
"blocks.7.mlp.proj.weight",
"blocks.7.mlp.proj.bias",
"blocks.8.ln_1.weight",
"blocks.8.ln_1.bias",
"blocks.8.ln_2.weight",
"blocks.8.ln_2.bias",
"blocks.8.attn.qkv.weight",
"blocks.8.attn.qkv.bias",
"blocks.8.attn.out_proj.weight",
"blocks.8.attn.out_proj.bias",
"blocks.8.mlp.fc.weight",
"blocks.8.mlp.fc.bias",
"blocks.8.mlp.proj.weight",
"blocks.8.mlp.proj.bias",
"blocks.9.ln_1.weight",
"blocks.9.ln_1.bias",
"blocks.9.ln_2.weight",
"blocks.9.ln_2.bias",
"blocks.9.attn.qkv.weight",
"blocks.9.attn.qkv.bias",
"blocks.9.attn.out_proj.weight",
"blocks.9.attn.out_proj.bias",
"blocks.9.mlp.fc.weight",
"blocks.9.mlp.fc.bias",
"blocks.9.mlp.proj.weight",
"blocks.9.mlp.proj.bias",
"blocks.10.ln_1.weight",
"blocks.10.ln_1.bias",
"blocks.10.ln_2.weight",
"blocks.10.ln_2.bias",
"blocks.10.attn.qkv.weight",
"blocks.10.attn.qkv.bias",
"blocks.10.attn.out_proj.weight",
"blocks.10.attn.out_proj.bias",
"blocks.10.mlp.fc.weight",
"blocks.10.mlp.fc.bias",
"blocks.10.mlp.proj.weight",
"blocks.10.mlp.proj.bias",
"blocks.11.ln_1.weight",
"blocks.11.ln_1.bias",
"blocks.11.ln_2.weight",
"blocks.11.ln_2.bias",
"blocks.11.attn.qkv.weight",
"blocks.11.attn.qkv.bias",
"blocks.11.attn.out_proj.weight",
"blocks.11.attn.out_proj.bias",
"blocks.11.mlp.fc.weight",
"blocks.11.mlp.fc.bias",
"blocks.11.mlp.proj.weight",
"blocks.11.mlp.proj.bias",
"blocks.12.ln_1.weight",
"blocks.12.ln_1.bias",
"blocks.12.ln_2.weight",
"blocks.12.ln_2.bias",
"blocks.12.attn.qkv.weight",
"blocks.12.attn.qkv.bias",
"blocks.12.attn.out_proj.weight",
"blocks.12.attn.out_proj.bias",
"blocks.12.mlp.fc.weight",
"blocks.12.mlp.fc.bias",
"blocks.12.mlp.proj.weight",
"blocks.12.mlp.proj.bias",
"blocks.13.ln_1.weight",
"blocks.13.ln_1.bias",
"blocks.13.ln_2.weight",
"blocks.13.ln_2.bias",
"blocks.13.attn.qkv.weight",
"blocks.13.attn.qkv.bias",
"blocks.13.attn.out_proj.weight",
"blocks.13.attn.out_proj.bias",
"blocks.13.mlp.fc.weight",
"blocks.13.mlp.fc.bias",
"blocks.13.mlp.proj.weight",
"blocks.13.mlp.proj.bias",
"blocks.14.ln_1.weight",
"blocks.14.ln_1.bias",
"blocks.14.ln_2.weight",
"blocks.14.ln_2.bias",
"blocks.14.attn.qkv.weight",
"blocks.14.attn.qkv.bias",
"blocks.14.attn.out_proj.weight",
"blocks.14.attn.out_proj.bias",
"blocks.14.mlp.fc.weight",
"blocks.14.mlp.fc.bias",
"blocks.14.mlp.proj.weight",
"blocks.14.mlp.proj.bias",
"blocks.15.ln_1.weight",
"blocks.15.ln_1.bias",
"blocks.15.ln_2.weight",
"blocks.15.ln_2.bias",
"blocks.15.attn.qkv.weight",
"blocks.15.attn.qkv.bias",
"blocks.15.attn.out_proj.weight",
"blocks.15.attn.out_proj.bias",
"blocks.15.mlp.fc.weight",
"blocks.15.mlp.fc.bias",
"blocks.15.mlp.proj.weight",
"blocks.15.mlp.proj.bias",
"blocks.16.ln_1.weight",
"blocks.16.ln_1.bias",
"blocks.16.ln_2.weight",
"blocks.16.ln_2.bias",
"blocks.16.attn.qkv.weight",
"blocks.16.attn.qkv.bias",
"blocks.16.attn.out_proj.weight",
"blocks.16.attn.out_proj.bias",
"blocks.16.mlp.fc.weight",
"blocks.16.mlp.fc.bias",
"blocks.16.mlp.proj.weight",
"blocks.16.mlp.proj.bias",
"blocks.17.ln_1.weight",
"blocks.17.ln_1.bias",
"blocks.17.ln_2.weight",
"blocks.17.ln_2.bias",
"blocks.17.attn.qkv.weight",
"blocks.17.attn.qkv.bias",
"blocks.17.attn.out_proj.weight",
"blocks.17.attn.out_proj.bias",
"blocks.17.mlp.fc.weight",
"blocks.17.mlp.fc.bias",
"blocks.17.mlp.proj.weight",
"blocks.17.mlp.proj.bias",
"blocks.18.ln_1.weight",
"blocks.18.ln_1.bias",
"blocks.18.ln_2.weight",
"blocks.18.ln_2.bias",
"blocks.18.attn.qkv.weight",
"blocks.18.attn.qkv.bias",
"blocks.18.attn.out_proj.weight",
"blocks.18.attn.out_proj.bias",
"blocks.18.mlp.fc.weight",
"blocks.18.mlp.fc.bias",
"blocks.18.mlp.proj.weight",
"blocks.18.mlp.proj.bias",
"blocks.19.ln_1.weight",
"blocks.19.ln_1.bias",
"blocks.19.ln_2.weight",
"blocks.19.ln_2.bias",
"blocks.19.attn.qkv.weight",
"blocks.19.attn.qkv.bias",
"blocks.19.attn.out_proj.weight",
"blocks.19.attn.out_proj.bias",
"blocks.19.mlp.fc.weight",
"blocks.19.mlp.fc.bias",
"blocks.19.mlp.proj.weight",
"blocks.19.mlp.proj.bias",
"blocks.20.ln_1.weight",
"blocks.20.ln_1.bias",
"blocks.20.ln_2.weight",
"blocks.20.ln_2.bias",
"blocks.20.attn.qkv.weight",
"blocks.20.attn.qkv.bias",
"blocks.20.attn.out_proj.weight",
"blocks.20.attn.out_proj.bias",
"blocks.20.mlp.fc.weight",
"blocks.20.mlp.fc.bias",
"blocks.20.mlp.proj.weight",
"blocks.20.mlp.proj.bias",
"blocks.21.ln_1.weight",
"blocks.21.ln_1.bias",
"blocks.21.ln_2.weight",
"blocks.21.ln_2.bias",
"blocks.21.attn.qkv.weight",
"blocks.21.attn.qkv.bias",
"blocks.21.attn.out_proj.weight",
"blocks.21.attn.out_proj.bias",
"blocks.21.mlp.fc.weight",
"blocks.21.mlp.fc.bias",
"blocks.21.mlp.proj.weight",
"blocks.21.mlp.proj.bias",
"blocks.22.ln_1.weight",
"blocks.22.ln_1.bias",
"blocks.22.ln_2.weight",
"blocks.22.ln_2.bias",
"blocks.22.attn.qkv.weight",
"blocks.22.attn.qkv.bias",
"blocks.22.attn.out_proj.weight",
"blocks.22.attn.out_proj.bias",
"blocks.22.mlp.fc.weight",
"blocks.22.mlp.fc.bias",
"blocks.22.mlp.proj.weight",
"blocks.22.mlp.proj.bias",
"blocks.23.ln_1.weight",
"blocks.23.ln_1.bias",
"blocks.23.ln_2.weight",
"blocks.23.ln_2.bias",
"blocks.23.attn.qkv.weight",
"blocks.23.attn.qkv.bias",
"blocks.23.attn.out_proj.weight",
"blocks.23.attn.out_proj.bias",
"blocks.23.mlp.fc.weight",
"blocks.23.mlp.fc.bias",
"blocks.23.mlp.proj.weight",
"blocks.23.mlp.proj.bias",
"ln_f.weight",
"ln_f.bias",
"head.weight",
"head.bias"
],
"tokenizer_bundle": {
"special_tokens_map.json": "/mnt/apps/llm-nanochat/hf_exports/1gpu-llm-medium-v2/special_tokens_map.json",
"tokenizer.json": "/mnt/apps/llm-nanochat/hf_exports/1gpu-llm-medium-v2/tokenizer.json",
"tokenizer_config.json": "/mnt/apps/llm-nanochat/hf_exports/1gpu-llm-medium-v2/tokenizer_config.json",
"tokenizer_meta.json": "/mnt/apps/llm-nanochat/hf_exports/1gpu-llm-medium-v2/tokenizer_meta.json"
},
"tokenizer_reference": {
"packed_dataset_config_path": null,
"tokenizer_dir": "/mnt/apps/llm-nanochat/tokenizers/tokenizer_20260515_en50it50_webwiki_stratified_500M",
"training_config_path": null
},
"transformers_config": {
"activation_function": "gelu",
"architecture": "gpt2",
"architectures": [
"GPT2LMHeadModel"
],
"attn_pdrop": 0.0,
"block_type": "gpt2_prelayernorm",
"causal_mask_mode": "buffered_upper_triangular",
"embd_pdrop": 0.0,
"init_strategy": "gpt2_std_0.02_residual_scale",
"layer_norm_epsilon": 1e-05,
"model_type": "gpt2",
"n_ctx": 2500,
"n_embd": 1024,
"n_head": 16,
"n_layer": 24,
"n_positions": 2500,
"norm_order": "preln",
"norm_type": "layernorm",
"positional_encoding": "learned_absolute",
"resid_pdrop": 0.0,
"tie_word_embeddings": true,
"use_cache": true,
"vocab_size": 32000
},
"transformers_export_notes": {
"dropped_lm_head_bias_for_gpt2_compat": true,
"dropped_lm_head_bias_max_abs": 1.3327739238739014,
"dropped_lm_head_bias_mean_abs": 0.046846117824316025
},
"transformers_safetensors_path": "/mnt/apps/llm-nanochat/hf_exports/1gpu-llm-medium-v2/model.safetensors"
}
|