{
  "format_version": 1,
  "stage": "accepted_layer_0",
  "accepted_layers": [
    0
  ],
  "config": {
    "feature_dim": 32,
    "memory_rank": 64,
    "dilations": [
      1,
      2,
      4,
      8,
      16,
      32,
      64,
      128
    ],
    "shifted_window": 8,
    "train_output_projection": true
  },
  "layer_reports": [
    {
      "layer": 0,
      "accepted": false,
      "steps": 275,
      "nmse": 0.048099592328071594,
      "cosine": 0.9809460043907166,
      "probe_nll": 2.957142174243927,
      "incremental_delta_nll": 0.04405015707015991,
      "cumulative_delta_nll": 0.04405015707015991,
      "round": 1
    },
    {
      "layer": 0,
      "accepted": false,
      "steps": 300,
      "nmse": 0.02023264765739441,
      "cosine": 0.991240918636322,
      "probe_nll": 2.9379348754882812,
      "incremental_delta_nll": 0.02484285831451416,
      "cumulative_delta_nll": 0.02484285831451416,
      "round": 2
    },
    {
      "layer": 0,
      "accepted": false,
      "steps": 275,
      "nmse": 0.01509437058120966,
      "cosine": 0.9934894442558289,
      "probe_nll": 2.933886468410492,
      "incremental_delta_nll": 0.020794451236724854,
      "cumulative_delta_nll": 0.020794451236724854,
      "round": 3
    },
    {
      "layer": 0,
      "accepted": true,
      "steps": 125,
      "nmse": 0.018665021285414696,
      "cosine": 0.9919254779815674,
      "probe_nll": 2.9270507097244263,
      "incremental_delta_nll": 0.01395869255065918,
      "cumulative_delta_nll": 0.01395869255065918,
      "round": 4
    }
  ]
}