File size: 1,532 Bytes
5e8bfe6 aff3464 5e8bfe6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 | {
"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"
]
}
|