{ "WEIGHT_DIMENSIONS": { "edge_index.dtype": "torch.int64", "edge_index.shape": [ 2, 186685 ], "edge_type.dtype": "torch.int64", "edge_type.shape": [ 186685 ] }, "_name_or_path": "./save/unicrs_rec/redial/model_best", "architectures": [ "UnicrsRec" ], "kgprompt_config": { "hidden_size": 768, "kg_info": { "num_entities": 31162, "num_relations": 25, "pad_entity_id": 31161 }, "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" }