| { |
| "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" |
| ] |
| } |
|
|
