Mentors4EDU's picture
Upload ntk_controller/stage2_538m.json with huggingface_hub
b5fe276 verified
Raw History Blame Contribute Delete
969 Bytes
{
"selected_gates": 4096.0,
"select_seconds": 0.0,
"train_seconds": 127.37424483099994,
"steps_completed": 240.0,
"loss_first": 1.602540373802185,
"loss_last": 2.130554437637329,
"task_loss_first": 1.602540373802185,
"task_loss_last": 2.130554437637329,
"hook_site": "layer_output",
"stopped_early": 0.0,
"retain_weight": 0.0,
"kl_to_base": 0.0,
"validation_best_nll": 1.9539643727595735,
"validation_best_step": 240.0,
"stage": "stage2_538m",
"checkpoint": "/workspace/RiemannModel/checkpoints/stage2_538m",
"train_examples": 19000,
"validation_examples": 1000,
"steps": 240,
"lr": 0.005,
"gates_requested": 4096,
"fit_seconds": 127.37734150886536,
"validation_nll_base": 1.9681743749192033,
"validation_nll_final": 1.9539643727595735,
"warm_start": {
"prior_gates": 4096,
"carried_gates": 4096,
"new_gates_for_grown_dims": 0,
"total_gates": 4096,
"prior_layers": 18,
"prior_hidden": 1024
}
}