{
  "config": {
    "context_tokens": 224,
    "encoder": "tinyx",
    "heads": 4,
    "layers": 2,
    "option_tokens": 96,
    "rank": 128,
    "width": 128
  },
  "format": "cua-s1",
  "format_version": 1,
  "metadata": {
    "batch_size": 64,
    "best_validation": {
      "ece": 0.018766840075841174,
      "examples": 2855,
      "nll": 0.51966637646463,
      "per_action": {
        "check": {
          "acc": 1.0,
          "n": 87
        },
        "click": {
          "acc": 0.9479166666666666,
          "n": 96
        },
        "fill": {
          "acc": 0.7756680731364276,
          "n": 1422
        },
        "skip": {
          "acc": 0.8712,
          "n": 1250
        }
      },
      "top1": 0.8301225900650024
    },
    "deterministic": false,
    "epochs": 4,
    "history": [
      {
        "epoch": 1,
        "train_nll": 2.060510016020325,
        "validation": {
          "ece": 0.05491860159418138,
          "examples": 2855,
          "nll": 1.748936438351713,
          "per_action": {
            "check": {
              "acc": 0.4367816091954023,
              "n": 87
            },
            "click": {
              "acc": 0.3958333333333333,
              "n": 96
            },
            "fill": {
              "acc": 0.06118143459915612,
              "n": 1422
            },
            "skip": {
              "acc": 0.9712,
              "n": 1250
            }
          },
          "top1": 0.4823117256164551
        }
      },
      {
        "epoch": 2,
        "train_nll": 1.3498624871227647,
        "validation": {
          "ece": 0.029052662350295577,
          "examples": 2855,
          "nll": 1.1316383522153945,
          "per_action": {
            "check": {
              "acc": 0.9770114942528736,
              "n": 87
            },
            "click": {
              "acc": 0.8854166666666666,
              "n": 96
            },
            "fill": {
              "acc": 0.4212376933895921,
              "n": 1422
            },
            "skip": {
              "acc": 0.8232,
              "n": 1250
            }
          },
          "top1": 0.6297723054885864
        }
      },
      {
        "epoch": 3,
        "train_nll": 0.7656327178699136,
        "validation": {
          "ece": 0.02394736194401048,
          "examples": 2855,
          "nll": 0.6415376866970296,
          "per_action": {
            "check": {
              "acc": 0.9770114942528736,
              "n": 87
            },
            "click": {
              "acc": 0.9791666666666666,
              "n": 96
            },
            "fill": {
              "acc": 0.7088607594936709,
              "n": 1422
            },
            "skip": {
              "acc": 0.8464,
              "n": 1250
            }
          },
          "top1": 0.7863397598266602
        }
      },
      {
        "epoch": 4,
        "train_nll": 0.4496109453594249,
        "validation": {
          "ece": 0.018766840075841174,
          "examples": 2855,
          "nll": 0.51966637646463,
          "per_action": {
            "check": {
              "acc": 1.0,
              "n": 87
            },
            "click": {
              "acc": 0.9479166666666666,
              "n": 96
            },
            "fill": {
              "acc": 0.7756680731364276,
              "n": 1422
            },
            "skip": {
              "acc": 0.8712,
              "n": 1250
            }
          },
          "top1": 0.8301225900650024
        }
      }
    ],
    "learning_rate": 0.002,
    "seed": 7,
    "task": "form-specialist-choice-classification",
    "warmup": 0.05,
    "workers": 0
  },
  "state_signature": "86e836863df8a0b0caa34ac5eab18bcd4fb44e420f83e271c96379fac0b0aec4"
}