File size: 588 Bytes
86d8d72 04f2c42 86d8d72 |
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 |
{
"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"
}
|