{
  "arm": "split",
  "flow_messages": true,
  "flow_records": false,
  "record_encoder": null,
  "record_split": true,
  "capacity": 0.25,
  "epoch": 5,
  "validation_link_ap": 0.9951015999857535,
  "nested_keys": [
    "model",
    "head"
  ],
  "numpy_keys": [],
  "source_checkpoint": "context_encoder.pt",
  "tensor_names": [
    "head.net.0.bias",
    "head.net.0.weight",
    "head.net.2.bias",
    "head.net.2.weight",
    "model.attention.in_proj_bias",
    "model.attention.in_proj_weight",
    "model.attention.out_proj.bias",
    "model.attention.out_proj.weight",
    "model.features.gap.cold",
    "model.features.gap.time.omega",
    "model.key.bias",
    "model.key.weight",
    "model.late.flow.marker",
    "model.late.flow.norm.bias",
    "model.late.flow.norm.weight",
    "model.late.flow.out.0.bias",
    "model.late.flow.out.0.weight",
    "model.late.flow.out.2.bias",
    "model.late.flow.out.2.weight",
    "model.memory.cell.bias_hh",
    "model.memory.cell.bias_ih",
    "model.memory.cell.weight_hh",
    "model.memory.cell.weight_ih",
    "model.memory.gap.omega",
    "model.memory.h0",
    "model.memory.message.0.bias",
    "model.memory.message.0.weight",
    "model.memory.message.2.bias",
    "model.memory.message.2.weight",
    "model.neighbour_gap.omega",
    "model.neighbour_role.weight",
    "model.no_reply",
    "model.out.0.bias",
    "model.out.0.weight",
    "model.out.2.bias",
    "model.out.2.weight",
    "model.query.bias",
    "model.query.weight",
    "model.role.weight"
  ]
}