File size: 3,027 Bytes
9ac6718 | 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 | {
"schema_version": 1,
"stage": "base",
"experiment_id": "Think.Unbounded-d32-v2mix-cont",
"branch": {
"parent_experiment_id": "Think.Unbounded-d32",
"parent_step": 5500,
"lr_schedule": "continue",
"load_optimizer": true
},
"datasets": {
"original": {
"adapter": "parquet_shards",
"repo": "jbduran/think-dataset-clean-1930s",
"revision": "45225d95bc15f942be3b4b344738cea1f66e3de8",
"validation_shard": 472,
"num_train_shards": 330,
"download_workers": 4
},
"midtrain_r21": {
"adapter": "parquet_shards",
"repo": "zachnorton03/think-midtrain",
"revision": "5250c497c8c72558a573dbcd216e26ea3c76e6e2",
"subfolder": "mixed/v2/ratio_21/data",
"validation_shard": 65,
"num_train_shards": 65,
"download_workers": 4
},
"midtrain_r45": {
"adapter": "parquet_shards",
"repo": "zachnorton03/think-midtrain",
"revision": "5250c497c8c72558a573dbcd216e26ea3c76e6e2",
"subfolder": "mixed/v2/ratio_45/data",
"validation_shard": 33,
"num_train_shards": 33,
"download_workers": 4
}
},
"mixture_schedule": {
"total_tokens": 20132659200,
"seed_data": 1234,
"max_epochs": 4,
"stages": [
{
"name": "base",
"start_tokens": 0,
"source": "original"
},
{
"name": "injection",
"start_tokens": 14092861440,
"source": "midtrain_r21"
},
{
"name": "decay_mix",
"start_tokens": 18119393280,
"source": "midtrain_r45"
}
]
},
"tokenizer": {
"mode": "reuse",
"source_experiment_id": "clean1930s-d24-r12-ctx4096-fulltok-v1",
"vocab_size": 32768
},
"pretokenize": {
"enabled": true,
"slack": 1.02,
"require_no_wrap": true,
"val_tokens": 20971520,
"shard_tokens": 100000000,
"tokenizer_threads": 8
},
"training": {
"depth": 32,
"scaling_params": 1677724672,
"target_param_data_ratio": 12,
"max_seq_len": 4096,
"window_pattern": "SSSL",
"muon_momentum": 0.9,
"device_batch_size": 2,
"total_batch_size": 2097152,
"fp8": true,
"fp8_recipe": "tensorwise",
"seed": 42,
"save_every": 500,
"eval_every": 250,
"eval_tokens": 2097152,
"core_metric_every": -1,
"sample_every": -1
},
"artifacts": {
"repo": "jbduran/think.nano",
"keep_local_checkpoints": 1
},
"wandb": {
"entity": "jbduran-thinkingmachinesncsu",
"project": "think.nano",
"name": "Think.Unbounded-d32-v2mix-cont",
"group": "clean1930s-d32",
"tags": [
"think-dataset-clean-1930s",
"d32",
"ratio12",
"ctx4096",
"sssl",
"fp8",
"muon-momentum-constant",
"muon-momentum-0.90",
"midtrain-schedule",
"mix-v2",
"0-21-45",
"midtrain-2.88ep",
"data-repair-continuation"
]
},
"config_fingerprint": "00af0bde578c1269",
"artifact_path": "experiments/Think.Unbounded-d32-v2mix-cont"
}
|