search

LARS-research / AdaProp

[KDD 2023] AdaProp: Learning Adaptive Propagation for Graph Neural Network based Knowledge Graph Reasoning
https://arxiv.org/pdf/2205.15319.pdf
19 stars 3 forks source link

The following problem occurs when I run the code and after debugging the problem is not solved. How can I fix it? #9

Closed xiaowaiwaiA closed 1 week ago

xiaowaiwaiA commented 1 week ago

The following problem occurs when I run the code and after debugging the problem is not solved. How can I fix it? 1

AndrewZhou924 commented 1 week ago

Hi, you should give the GPU index, e.g., --gpu 0.

xiaowaiwaiA commented 1 week ago

Thank you for your reply. I followed your instructions to execute the code, but it seems that it cannot accept the parameter --gpu 0. 2

yzhangee commented 1 week ago

You can directly set the gpu number at https://github.com/LARS-research/AdaProp/blob/159715c9950e67cea432859f9381cea617348a16/inductive/train.py#L32 as 0.

xiaowaiwaiA commented 1 week ago

Thank you very much! I've taken up too much of your time.