Upload config.json with huggingface_hub

config.json

@@ -0,0 +1,76 @@
+{
+  "step": 464,
+  "val_bpb": 0.8649741211301406,
+  "model_config": {
+    "sequence_len": 2048,
+    "vocab_size": 32768,
+    "n_layer": 18,
+    "n_head": 9,
+    "n_kv_head": 9,
+    "n_embd": 1152,
+    "window_pattern": "L",
+    "use_mla": false,
+    "kv_lora_rank": 512,
+    "qk_lora_rank": 1536,
+    "qk_nope_head_dim": 128,
+    "qk_rope_head_dim": 64,
+    "v_head_dim": 128,
+    "use_ssa": true
+  },
+  "user_config": {
+    "run": "nanochat_p2",
+    "device_type": "",
+    "fp8": false,
+    "fp8_recipe": "tensorwise",
+    "depth": 18,
+    "aspect_ratio": 64,
+    "head_dim": 128,
+    "max_seq_len": 2048,
+    "window_pattern": "L",
+    "no_muon": false,
+    "mla": false,
+    "ssa": true,
+    "kv_lora_rank": 512,
+    "qk_lora_rank": 1536,
+    "qk_nope_head_dim": 128,
+    "qk_rope_head_dim": 64,
+    "v_head_dim": 128,
+    "num_iterations": -1,
+    "target_flops": -1.0,
+    "target_param_data_ratio": 1.5,
+    "device_batch_size": 32,
+    "total_batch_size": -1,
+    "embedding_lr": 0.3,
+    "unembedding_lr": 0.004,
+    "weight_decay": 0.2,
+    "matrix_lr": 0.02,
+    "adam_beta1": 0.8,
+    "adam_beta2": 0.95,
+    "warmup_ratio": 0.1,
+    "warmdown_ratio": 0.5,
+    "final_lr_frac": 0.0,
+    "resume_from_step": -1,
+    "eval_every": 250,
+    "eval_tokens": 20971520,
+    "core_metric_every": 2000,
+    "core_metric_max_per_task": 500,
+    "sample_every": 2000,
+    "save_every": -1,
+    "model_tag": "picochat_long",
+    "load_weights_from": "./nanochat_run/base_checkpoints/picochat",
+    "load_weights_step": null
+  },
+  "device_batch_size": 32,
+  "max_seq_len": 2048,
+  "total_batch_size": 1048576,
+  "dataloader_state_dict": {
+    "pq_idx": 12,
+    "rg_idx": 22,
+    "epoch": 1
+  },
+  "loop_state": {
+    "min_val_bpb": 0.8649741211301406,
+    "smooth_train_loss": 2.811691184728465,
+    "total_training_time": 6117.183136701584
+  }
+}