{
  "WEIGHT_DIMENSIONS": {
    "edge_index.dtype": "torch.int64",
    "edge_index.shape": [
      2,
      358184
    ],
    "edge_type.dtype": "torch.int64",
    "edge_type.shape": [
      358184
    ],
    "item_index.dtype": "torch.int64",
    "item_index.shape": [
      6924
    ]
  },
  "_name_or_path": "/home/zhankui/0_ongoing/RecBot/src/recwizard/modules/kbrd/temp_kbrd_rec",
  "architectures": [
    "KBRDRec"
  ],
  "dim": 128,
  "n_entity": 64368,
  "n_relation": 214,
  "num_bases": 8,
  "sub_n_relation": 35,
  "torch_dtype": "float32",
  "transformers_version": "4.33.3"
}