nanochat-d20 / report /report.md
sampathchanda's picture
Upload folder using huggingface_hub
a465972 verified

nanochat training report

Generated: 2025-10-13 18:24:03

Environment

Git Information

  • Branch: master
  • Commit: 626bd3e (clean)
  • Message: Add image of the WebUI to readme

Hardware

  • Platform: Linux
  • CPUs: 48 cores (96 logical)
  • Memory: 1121.8 GB
  • GPUs: 8x NVIDIA A100-SXM4-80GB
  • GPU Memory: 634.0 GB total
  • CUDA Version: 12.8
  • Hourly Rate: $14.32/hour

Software

  • Python: 3.10.12
  • PyTorch: 2.8.0+cu128

Bloat

  • Characters: 330,622
  • Lines: 8,077
  • Files: 42
  • Tokens (approx): 82,655
  • Dependencies (uv.lock lines): 2,004

Run started: 2025-10-13 18:24:07


Tokenizer training

timestamp: 2025-10-13 18:25:52

  • max_chars: 2,000,000,000
  • doc_cap: 10,000
  • vocab_size: 65,536
  • train_time: 89.5722
  • num_special_tokens: 9
  • token_bytes_min: 1
  • token_bytes_max: 32
  • token_bytes_mean: 6.9151
  • token_bytes_std: 2.8736

Tokenizer evaluation

timestamp: 2025-10-13 18:26:00

Comparison with GPT-2

Text Type Bytes GPT-2 Tokens GPT-2 Ratio Ours Tokens Ours Ratio Relative Diff %
news 1819 404 4.50 375 4.85 +7.2%
korean 893 745 1.20 721 1.24 +3.2%
code 1259 576 2.19 493 2.55 +14.4%
math 1834 936 1.96 966 1.90 -3.2%
science 1112 260 4.28 225 4.94 +13.5%
fwe-train 4208518 900364 4.67 856901 4.91 +4.8%
fwe-val 4908443 1059062 4.63 1010356 4.86 +4.6%

Comparison with GPT-4

Text Type Bytes GPT-4 Tokens GPT-4 Ratio Ours Tokens Ours Ratio Relative Diff %
news 1819 387 4.70 375 4.85 +3.1%
korean 893 364 2.45 721 1.24 -98.1%
code 1259 309 4.07 493 2.55 -59.5%
math 1834 832 2.20 966 1.90 -16.1%
science 1112 249 4.47 225 4.94 +9.6%
fwe-train 4208518 874799 4.81 856901 4.91 +2.0%
fwe-val 4908443 1029691 4.77 1010356 4.86 +1.9%

Base model training

timestamp: 2025-10-14 01:32:23

  • run: d0
  • depth: 20
  • max_seq_len: 2048
  • num_iterations: -1
  • target_flops: -1.0000
  • target_param_data_ratio: 20
  • device_batch_size: 32
  • total_batch_size: 524,288
  • embedding_lr: 0.2000
  • unembedding_lr: 0.0040
  • weight_decay: 0.0000
  • matrix_lr: 0.0200
  • grad_clip: 1.0000
  • eval_every: 250
  • eval_tokens: 10,485,760
  • core_metric_every: 2000
  • core_metric_max_per_task: 500
  • sample_every: 2000
  • model_tag:
  • Number of parameters: 560,988,160
  • Number of FLOPs per token: 3.491758e+09
  • Calculated number of iterations: 21,400
  • Number of training tokens: 11,219,763,200
  • Tokens : Params ratio: 20.0000
  • DDP world size: 8
  • warmup_ratio: 0.0000
  • warmdown_ratio: 0.2000
  • final_lr_frac: 0.0000
  • Minimum validation bpb: 0.8149
  • Final validation bpb: 0.8149
  • CORE metric estimate: 0.2059
  • MFU %: 21.08%
  • Total training flops: 3.917670e+19
  • Total training time: 393.81m
  • Peak memory usage: 75374.27MiB

Base model loss

timestamp: 2025-10-14 01:34:11

  • train bpb: 0.8178
  • val bpb: 0.8150
  • sample 0: <|bos|>The capital of France is Paris. It is the largest city in France and the second largest in Europe.
  • sample 1: <|bos|>The chemical symbol of gold is Au. It is a soft, malleable, ductile, and malleable metal. It
  • sample 2: <|bos|>If yesterday was Friday, then tomorrow will be Saturday. If tomorrow is Sunday, then tomorrow will be Monday. If tomorrow is
  • sample 3: <|bos|>The opposite of hot is cold. The opposite of cold is hot. The opposite of hot is cold.
  • sample 4: <|bos|>The planets of the solar system are: Mercury, Venus, Earth, Mars, Jupiter, Saturn, Uranus, Neptune,
  • sample 5: <|bos|>My favorite color is red. I love the color red. I love the color red. I love
  • sample 6: <|bos|>If 5x + 3 = 13, then x is 5 times 3. If 5x + 3 =

Base model evaluation

timestamp: 2025-10-14 01:41:37

  • Model: base_model (step 21400)
  • CORE metric: 0.1976
  • hellaswag_zeroshot: 0.2598
  • jeopardy: 0.0874
  • bigbench_qa_wikidata: 0.5113
  • arc_easy: 0.5354
  • arc_challenge: 0.1183
  • copa: 0.2800
  • commonsense_qa: 0.0796
  • piqa: 0.3798
  • openbook_qa: 0.1627
  • lambada_openai: 0.3839
  • hellaswag: 0.2595
  • winograd: 0.2821
  • winogrande: 0.0513
  • bigbench_dyck_languages: 0.1430
  • agi_eval_lsat_ar: 0.1304
  • bigbench_cs_algorithms: 0.3727
  • bigbench_operators: 0.1762
  • bigbench_repeat_copy_logic: 0.0312
  • squad: 0.2389
  • coqa: 0.2088
  • boolq: -0.5218
  • bigbench_language_identification: 0.1757

Midtraining

timestamp: 2025-10-14 02:01:41

  • run: d0
  • dtype: bfloat16
  • max_seq_len: 2048
  • device_batch_size: 32
  • unembedding_lr: 0.0040
  • embedding_lr: 0.2000
  • matrix_lr: 0.0200
  • init_lr_frac: 1.0000
  • weight_decay: 0.0000
  • final_lr_frac: 0.0000
  • eval_every: 150
  • eval_tokens: 10,485,760
  • total_batch_size: 524,288
  • Number of iterations: 765
  • DDP world size: 8
  • Minimum validation bpb: 0.4176

Chat evaluation mid

timestamp: 2025-10-14 02:16:06

  • source: mid
  • task_name: None
  • dtype: bfloat16
  • temperature: 0.0000
  • max_new_tokens: 512
  • num_samples: 1
  • top_k: 50
  • batch_size: 8
  • model_tag: None
  • step: None
  • max_problems: None
  • ARC-Easy: 0.3758
  • ARC-Challenge: 0.2884
  • MMLU: 0.3088
  • GSM8K: 0.0303
  • HumanEval: 0.0671
  • ChatCORE metric: 0.0790

Chat SFT

timestamp: 2025-10-14 02:27:42

  • run: d0
  • source: mid
  • dtype: bfloat16
  • device_batch_size: 4
  • num_epochs: 1
  • max_iterations: -1
  • target_examples_per_step: 32
  • unembedding_lr: 0.0040
  • embedding_lr: 0.2000
  • matrix_lr: 0.0200
  • weight_decay: 0.0000
  • init_lr_frac: 0.0200
  • eval_every: 100
  • eval_steps: 100
  • eval_metrics_every: 200
  • Training rows: 20,843
  • Number of iterations: 651
  • Training loss: 1.2206
  • Validation loss: 1.0725

Chat evaluation sft

timestamp: 2025-10-14 02:39:37

  • source: sft
  • task_name: None
  • dtype: bfloat16
  • temperature: 0.0000
  • max_new_tokens: 512
  • num_samples: 1
  • top_k: 50
  • batch_size: 8
  • model_tag: None
  • step: None
  • max_problems: None
  • ARC-Easy: 0.3952
  • ARC-Challenge: 0.2961
  • MMLU: 0.3138
  • GSM8K: 0.0402
  • HumanEval: 0.0549
  • ChatCORE metric: 0.0870

Summary

  • Characters: 330,622
  • Lines: 8,077
  • Files: 42
  • Tokens (approx): 82,655
  • Dependencies (uv.lock lines): 2,004
Metric BASE MID SFT RL
CORE 0.1976 - - -
ARC-Challenge - 0.2884 0.2961 -
ARC-Easy - 0.3758 0.3952 -
GSM8K - 0.0303 0.0402 -
HumanEval - 0.0671 0.0549 -
MMLU - 0.3088 0.3138 -
ChatCORE - 0.0790 0.0870 -

Total wall clock time: 8h15m