infon-reasoner/onnx/meta.json

{
  "hidden_dim": 64,
  "n_layers": 2,
  "situation_dim": 16,
  "n_relations": 11,
  "relation_types": [
    "INITIATES",
    "ASSERTS",
    "TARGETS",
    "NEXT",
    "ENTAILS",
    "LOCATED_AT",
    "CONTRADICTS",
    "CAUSES",
    "SAME_REFERENT",
    "DEFINED_AS",
    "REFERS_TO"
  ],
  "backbone": {
    "file": "backbone.onnx",
    "inputs": [
      "node_features",
      "edge_index",
      "edge_types",
      "edge_weights",
      "situation_features"
    ],
    "output": "refined_node_features",
    "hidden_dim": 64,
    "n_layers": 2,
    "situation_dim": 16,
    "n_relations": 11,
    "fp32_mb": 0.4173,
    "out_shape_at_n9": [
      9,
      64
    ],
    "fp16_file": "backbone.fp16.onnx",
    "fp16_mb": 0.228
  },
  "operators": {
    "not": {
      "file": "ikl_not.onnx",
      "inputs": [
        "m"
      ],
      "output": "m_out",
      "fp32_mb": 0.0377,
      "kind": "unary",
      "hidden_dim": 64,
      "out_shape": [
        2,
        4
      ],
      "fp16_file": "ikl_not.fp16.onnx",
      "fp16_mb": 0.0197,
      "fp16_out_shape": [
        2,
        4
      ]
    },
    "that": {
      "file": "ikl_that.onnx",
      "inputs": [
        "m"
      ],
      "output": "m_out",
      "fp32_mb": 0.0376,
      "kind": "unary",
      "hidden_dim": 64,
      "out_shape": [
        2,
        4
      ],
      "fp16_file": "ikl_that.fp16.onnx",
      "fp16_mb": 0.0196,
      "fp16_out_shape": [
        2,
        4
      ]
    },
    "and": {
      "file": "ikl_and.onnx",
      "inputs": [
        "m_set"
      ],
      "output": "m_out",
      "fp32_mb": 0.0392,
      "kind": "reducer",
      "hidden_dim": 64,
      "out_shape": [
        2,
        4
      ],
      "fp16_file": "ikl_and.fp16.onnx",
      "fp16_mb": 0.0218,
      "fp16_out_shape": [
        2,
        4
      ]
    },
    "or": {
      "file": "ikl_or.onnx",
      "inputs": [
        "m_set"
      ],
      "output": "m_out",
      "fp32_mb": 0.0392,
      "kind": "reducer",
      "hidden_dim": 64,
      "out_shape": [
        2,
        4
      ],
      "fp16_file": "ikl_or.fp16.onnx",
      "fp16_mb": 0.0218,
      "fp16_out_shape": [
        2,
        4
      ]
    },
    "forall": {
      "file": "ikl_forall.onnx",
      "inputs": [
        "m_set"
      ],
      "output": "m_out",
      "fp32_mb": 0.0392,
      "kind": "reducer",
      "hidden_dim": 64,
      "out_shape": [
        2,
        4
      ],
      "fp16_file": "ikl_forall.fp16.onnx",
      "fp16_mb": 0.0218,
      "fp16_out_shape": [
        2,
        4
      ]
    },
    "exists": {
      "file": "ikl_exists.onnx",
      "inputs": [
        "m_set"
      ],
      "output": "m_out",
      "fp32_mb": 0.0392,
      "kind": "reducer",
      "hidden_dim": 64,
      "out_shape": [
        2,
        4
      ],
      "fp16_file": "ikl_exists.fp16.onnx",
      "fp16_mb": 0.0218,
      "fp16_out_shape": [
        2,
        4
      ]
    },
    "if": {
      "file": "ikl_if.onnx",
      "inputs": [
        "m1",
        "m2"
      ],
      "output": "m_out",
      "fp32_mb": 0.074,
      "kind": "binary",
      "hidden_dim": 64,
      "out_shape": [
        2,
        4
      ],
      "fp16_file": "ikl_if.fp16.onnx",
      "fp16_mb": 0.0386,
      "fp16_out_shape": [
        2,
        4
      ]
    },
    "iff": {
      "file": "ikl_iff.onnx",
      "inputs": [
        "m1",
        "m2"
      ],
      "output": "m_out",
      "fp32_mb": 0.1084,
      "kind": "binary",
      "hidden_dim": 64,
      "out_shape": [
        2,
        4
      ],
      "fp16_file": "ikl_iff.fp16.onnx",
      "fp16_mb": 0.0572,
      "fp16_out_shape": [
        2,
        4
      ]
    },
    "ist": {
      "file": "ikl_ist.onnx",
      "inputs": [
        "m",
        "sit"
      ],
      "output": "m_out",
      "fp32_mb": 0.0255,
      "kind": "ist",
      "hidden_dim": 64,
      "out_shape": [
        2,
        4
      ],
      "fp16_file": "ikl_ist.fp16.onnx",
      "fp16_mb": 0.0137,
      "fp16_out_shape": [
        2,
        4
      ]
    }
  },
  "heads": {
    "next_anchor": {
      "file": "head_next_anchor.onnx",
      "inputs": [
        "node_features"
      ],
      "output": "probs",
      "n_classes": 1,
      "fp32_mb": 0.0006,
      "fp16_file": "head_next_anchor.fp16.onnx",
      "fp16_mb": 0.0008
    },
    "anomaly": {
      "file": "head_anomaly.onnx",
      "inputs": [
        "node_features"
      ],
      "output": "anomaly_score",
      "fp32_mb": 0.0096,
      "fp16_file": "head_anomaly.fp16.onnx",
      "fp16_mb": 0.0058
    },
    "role_type": {
      "file": "head_role_type.onnx",
      "inputs": [
        "h_a",
        "h_b",
        "mask_onehot"
      ],
      "output": "logits",
      "n_classes": 3,
      "role_types": [
        "actor",
        "feature",
        "relation"
      ],
      "fp32_mb": 0.0354,
      "fp16_file": "head_role_type.fp16.onnx",
      "fp16_mb": 0.0187
    }
  }
}