AStarNet
AStarNet copied to clipboard
Official implementation of A* Networks
Hi, Thanks for sharing the code for A*NET. While installing packages like torchdrug it is resulting in different types of errors. There is a conflict for the packages. Could you...
```bash (astarnet) [AStarNet]$ python3 script/run.py -c config/transductive/fb15k237_astarnet.yaml --gpus [0] 12:22:42 Config file: config/transductive/fb15k237_astarnet.yaml 12:22:42 {'output_dir': '~/experiments/', 'dataset': {'class': 'FB15k237', 'path': '~/datasets/knowledge_graphs/'}, 'task': {'class': 'KnowledgeGraphCompletion', 'model': {'class': 'AStarNet', 'base_layer': {'class': 'NBFNetConv',...
Hi, Thanks for sharing the codes for A*Net. The A*Net is trained and validated successfully but an error occurred in loading the checkpoint. Here are the details: 10:02:04 Load checkpoint...
`Loading /public/home/yqzhang/modeltest/0_path_model/AStarNet-master/datasets/knowledge_graphs/fb15k237_train.tx Loading /public/home/yqzhang/modeltest/0_path_model/AStarNet-master/datasets/knowledge_graphs/fb15k237_valid.tx Loading /public/home/yqzhang/modeltest/0_path_model/AStarNet-master/datasets/knowledge_graphs/fb15k237_test.txt 10:25:25 FB15k237( #entity: 14541 #relation: 237 #triplet: 310116 ) 10:25:25 #train: 272115, #valid: 17535, #test: 20466 Traceback (most recent call last): File "/public/home/yqzhang/anaconda3/envs/kgmodel/lib/python3.7/site-packages/torch/utils/cpp_extension.py", line...
Traceback (most recent call last): File "/home/nudt/linyawei/AStarNet/script/visualize.py", line 175, in visualize(solver, solver.test_set[i], entity_vocab, relation_vocab, output_file="/home/nudt/linyawei/AStarNet/results.txt") File "/home/nudt/linyawei/AStarNet/script/visualize.py", line 112, in visualize paths, weights, num_steps = solver.model.visualize(vis_batch) File "/home/nudt/anaconda3/envs/astar/lib/python3.10/site-packages/torchdrug-0.2.1-py3.10.egg/torchdrug/tasks/reasoning.py", line 222,...