Hugging Face's logo

Kevin99z
/
UnicrsRec

Transformers
PyTorch
Inference Endpoints
Model card Files Community
UnicrsRec
File size: 785 Bytes 
6f53581
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
 
{
  "WEIGHT_DIMENSIONS": {
    "edge_index.dtype": "torch.int64",
    "edge_index.shape": [
      2,
      69125
    ],
    "edge_type.dtype": "torch.int64",
    "edge_type.shape": [
      69125
    ]
  },
  "_name_or_path": "./save/unicrs_rec/inspired/model_best",
  "architectures": [
    "UnicrsRec"
  ],
  "kgprompt_config": {
    "hidden_size": 768,
    "kg_info": {
      "num_entities": 17321,
      "num_relations": 13,
      "pad_entity_id": 17320
    },
    "n_block": 2,
    "n_head": 12,
    "n_layer": 12,
    "n_prefix_rec": 10,
    "num_bases": 8,
    "num_tokens": 50266,
    "token_hidden_size": 768
  },
  "num_tokens": 50259,
  "pad_token_id": 50257,
  "pretrained_model": "microsoft/DialoGPT-small",
  "torch_dtype": "float32",
  "transformers_version": "4.29.2"
}