Mentors4EDU's picture
Upload NTK controller summary (stage6_12.85b)
6153f77 verified
Raw History Blame Contribute Delete
980 Bytes
{
"selected_gates": 6732.0,
"select_seconds": 0.0,
"train_seconds": 2026.6077014809998,
"steps_completed": 240.0,
"loss_first": 0.9671240448951721,
"loss_last": 1.4579845666885376,
"task_loss_first": 0.9671240448951721,
"task_loss_last": 1.4579845666885376,
"hook_site": "layer_output",
"stopped_early": 0.0,
"retain_weight": 0.0,
"kl_to_base": 0.0,
"validation_best_nll": 1.3103517218016936,
"validation_best_step": 240.0,
"stage": "stage6_12.85b",
"checkpoint": "/workspace/RiemannModel/checkpoints/stage6_12.85b",
"train_examples": 19000,
"validation_examples": 1000,
"steps": 240,
"lr": 0.005,
"gates_requested": 8192,
"fit_seconds": 2026.6114346981049,
"validation_nll_base": 1.3227896775331283,
"validation_nll_final": 1.3103517218016936,
"warm_start": {
"prior_gates": 4102,
"carried_gates": 4102,
"new_gates_for_grown_dims": 2630,
"total_gates": 6732,
"prior_layers": 32,
"prior_hidden": 4096
}
}