nadir / config.json
ProCreations's picture
Nadir v2: element head, inner-page data, runtime fixes
864f427 verified
Raw History Blame Contribute Delete
998 Bytes
{
"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
}
}