{
  "arm": "split",
  "flow_messages": true,
  "flow_records": true,
  "record_encoder": "/home/singh/netwatch-data/runs/20260926-2337/record_encoder.pt",
  "record_split": true,
  "capacity": 0.25,
  "epoch": 5,
  "validation_link_ap": 0.9954236793562934,
  "nested_keys": [
    "model",
    "head"
  ],
  "numpy_keys": [],
  "source_checkpoint": "world_model_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.late.record.encoder.flow_in.0.bias",
    "model.late.record.encoder.flow_in.0.weight",
    "model.late.record.encoder.flow_in.2.bias",
    "model.late.record.encoder.flow_in.2.weight",
    "model.late.record.encoder.flow_norm.bias",
    "model.late.record.encoder.flow_norm.weight",
    "model.late.record.encoder.out.0.bias",
    "model.late.record.encoder.out.0.weight",
    "model.late.record.encoder.out.2.bias",
    "model.late.record.encoder.out.2.weight",
    "model.late.record.encoder.side.weight",
    "model.late.record.encoder.side_in.0.bias",
    "model.late.record.encoder.side_in.0.weight",
    "model.late.record.encoder.side_in.2.bias",
    "model.late.record.encoder.side_in.2.weight",
    "model.late.record.encoder.side_norm.bias",
    "model.late.record.encoder.side_norm.weight",
    "model.late.record.marker",
    "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"
  ]
}