{
 "d": 384,
 "layers": 8,
 "heads": 6,
 "ff": 4,
 "drop": 0.1,
 "architecture": "NadirNet",
 "version": 2,
 "token_features": 49,
 "image_features": 41,
 "relation_channels": 11,
 "max_tokens": 256,
 "max_images": 32,
 "page_graph": {
  "inputs": [
   "tok_f",
   "tok_role",
   "tok_lab",
   "img_f",
   "img_thumb",
   "page_f",
   "src",
   "rel",
   "mask"
  ],
  "outputs": [
   "lab",
   "alpha_logit",
   "ctx"
  ]
 },
 "element_graph": {
  "inputs": [
   "e_f",
   "e_thumb",
   "under_light",
   "under_dark",
   "ctx"
  ],
  "outputs": [
   "treat_logits",
   "role_logits"
  ]
 },
 "element_treatments": [
  "keep",
  "invert",
  "adapt",
  "remove"
 ],
 "image_roles": [
  "photo",
  "illustration",
  "diagram",
  "logo",
  "icon",
  "text",
  "backdrop",
  "avatar",
  "ui"
 ],
 "context_dim": 32,
 "element_temperature": 1.3000000000000007,
 "element_calibration": {
  "temperature": 1.3000000000000007,
  "nll": 0.32924941182136536,
  "ece": 0.028595900163054466,
  "n": 23532
 }
}