d1a-e2b-it / training_config.json
JohnP1's picture
D1A-E2B-it v0.6
1d14f1a verified
Raw History Blame Contribute Delete
2 kB
{
"args": {
"base": "google/gemma-4-E2B-it",
"n_per_source": 1000,
"epochs": 1,
"lr": 5e-05,
"head_lr": 0.0,
"weight_decay": 0.01,
"lora": 16,
"accum": 4,
"holdout": "",
"suite": "evals/v7/decision-v7",
"train_sources": "",
"device": "cuda",
"batch": 2,
"dtype": "bf16",
"weights_dtype": "bf16",
"checkpointing": 1,
"head_dim": 256,
"lora_targets": "all",
"base_revision": "3e22461f65e89153144f8adb70e3b8c2cc9845a7",
"p_none": 0.1,
"p_none_distract": 0.12,
"p_distract": 0.15,
"p_none_pair": 0.0,
"none_pair_max_state": null,
"synthetic_repeat": 1,
"public_frac": 1.0,
"out": "e2b-it/stage2",
"data": "data/train-e2b-v06.jsonl",
"max_state": 5120,
"extra_suites": "",
"replay": 0,
"init_from": "e2b-it/stage1",
"allow_missing_sources": "",
"reason": "",
"row_budget": 0,
"shared_prefix": 0,
"length_sort": 0,
"pass_tokens_max": 0,
"max_steps": 0,
"save_every_steps": 0,
"save_every_minutes": 30.0,
"resume": 1,
"stop_after": 0,
"seed": 0
},
"suite_sha256": "a8f50e481b7d90b97da049e0ff6a01cee2f1ed204aed61a8265af0edbb5514d2",
"base_revision": "3e22461f65e89153144f8adb70e3b8c2cc9845a7",
"init_source": {
"init_from": "e2b-it/stage1",
"resolved": "e2b-it/stage1",
"weights_sha256": "ce9d3ac952af9c1045cea80e981bd0612b3efb9c199a308616886f77946bdcb7",
"head_sha256": "d4e3e8d0f07ee1b24c7fcca957052d91ea391b92ffb1aeddc88f838c70581806",
"tensors": 410
},
"holdout": [],
"sources": {
"covered": [
"evals/hard-v1",
"evals/devtools-v1",
"evals/documents-v1",
"evals/d1a/ja-jglue:train",
"evals/d1a/pr-labels:train",
"evals/d1a/pr-labels:train-ja",
"evals/d1a/pr-labels:train-blast",
"evals/d1a/routing:factory-train",
"evals/d1a/routing:generic-train",
"evals/v7/decision-v7"
],
"allowed_missing": [],
"reason": null
}
}