| { |
| "data_dir": "/dev/shm/data/fineweb-edu-25B", |
| "batch_size": 8, |
| "seq_len": 1024, |
| "max_iters": 2730, |
| "gradient_accumulation_steps": 192, |
| "seed": 1337, |
| "optimizer": "muon", |
| "learning_rate": 0.0006, |
| "min_lr": 6e-05, |
| "weight_decay": 0.1, |
| "beta1": 0.9, |
| "beta2": 0.95, |
| "grad_clip": 1.0, |
| "warmup_iters": 200, |
| "lr_decay_iters": 2730, |
| "muon_lr": 0.012, |
| "muon_min_lr": 0.00015, |
| "adamw_lr": 0.012, |
| "adamw_min_lr": 0.00015, |
| "muon_momentum": 0.95, |
| "muon_nesterov": true, |
| "muon_ns_steps": 5, |
| "muon_weight_decay": 0.0, |
| "lr_schedule": "linear_warmdown", |
| "warmdown_frac": 0.3, |
| "parity_target": 0.0, |
| "parity_margin": 0.03, |
| "parity_warmdown_iters": 1500, |
| "parity_trigger_iter": -1, |
| "retain_from_iter": -1, |
| "save_at_val": [], |
| "device": "cuda", |
| "dtype": "bfloat16", |
| "compile": true, |
| "n_layer": 12, |
| "n_head": 8, |
| "n_embd": 1024, |
| "block_size": 1024, |
| "vocab_size": 50304, |
| "pos_encoding": "learned", |
| "rope_theta": 10000.0, |
| "sparsity_mode": "topk", |
| "rmsnorm_affine": true, |
| "topk_dict_size": 65536, |
| "topk_k": 48, |
| "cayley_levels": [], |
| "cayley_per_parent_budget": false, |
| "cayley_seed": 42, |
| "cayley_backend": "auto", |
| "cayley_locations": [ |
| "mlp_in" |
| ], |
| "cayley_score_standardize": false, |
| "cayley_forward_standardized": false, |
| "cayley_score_std_ema_beta": 0.99, |
| "cayley_replaces_pre_rmsnorm": false, |
| "cayley_richardson_refine": false, |
| "cayley_matching_pursuit": false, |
| "tied_block_init": false, |
| "dead_threshold_c": 0.1, |
| "resume": true, |
| "out_dir": "/dev/shm/out/topk-small-65k-wallclock-matched", |
| "eval_interval": 250, |
| "eval_iters": 100, |
| "log_interval": 10, |
| "wandb_project": "sparse-nanogpt", |
| "wandb_run_name": "topk-small-65k-wallclock-matched", |
| "wandb_step_offset": 0, |
| "quick_eval_interval": 0, |
| "quick_eval_timeout": 900, |
| "hierarchy_eval_enabled": true, |
| "hierarchy_eval_iters": null |
| } |
