Upload train_stats.json with huggingface_hub

train_stats.json

@@ -0,0 +1,29 @@
+{
+ "base": "google/gemma-4-31B-it",
+ "method": "fsdp2-custom-loop",
+ "n_rows": 41951,
+ "n_self": 0,
+ "n_sequences": 20409,
+ "n_tokens": 32716533,
+ "world": 8,
+ "micro_batch": 1,
+ "accum": 8,
+ "steps": 318,
+ "lr": 1e-05,
+ "optim": "AdamW fused, fp32 master, bf16 compute, fp32 grad reduce",
+ "loss_curve": [
+  1.4239871501922607,
+  1.5003830194473267,
+  1.345046043395996,
+  1.3999685049057007
+ ],
+ "loss_first": [
+  1.4239871501922607
+ ],
+ "loss_last": [
+  1.3999685049057007
+ ],
+ "train_seconds": 547.8836672306061,
+ "tokens_per_second": 59714.37908593377,
+ "resumed_from": "/workspace/sdf/out/ctrl-SG/ckpt-step300"
+}