{
  "selected_gates": 4102.0,
  "select_seconds": 0.0,
  "train_seconds": 972.813525182999,
  "steps_completed": 240.0,
  "loss_first": 1.0246328115463257,
  "loss_last": 1.6060656309127808,
  "task_loss_first": 1.0246328115463257,
  "task_loss_last": 1.6060656309127808,
  "hook_site": "layer_output",
  "stopped_early": 0.0,
  "retain_weight": 0.0,
  "kl_to_base": 0.0,
  "validation_best_nll": 1.4470029361840933,
  "validation_best_step": 240.0,
  "stage": "stage5_6.66b",
  "checkpoint": "/workspace/RiemannModel/checkpoints/stage5_6.66b",
  "train_examples": 19000,
  "validation_examples": 1000,
  "steps": 240,
  "lr": 0.005,
  "gates_requested": 8192,
  "fit_seconds": 972.8165621757507,
  "validation_nll_base": 1.4513602711846518,
  "validation_nll_final": 1.4470029361840933,
  "warm_start": {
    "prior_gates": 4102,
    "carried_gates": 4102,
    "new_gates_for_grown_dims": 0,
    "total_gates": 4102,
    "prior_layers": 32,
    "prior_hidden": 2560
  }
}