mrtree777 / grf-gat Goto Github PK

作者您好，很荣幸能看到您这篇论文，我认真阅读了论文，并且尝试在自己数据集上使用这个模型，但是报了下面这个错误
File "/ tmp/msf-tl/modeLs/GATConv.py", Line 68， in forward
x = seLf .conv1(x = x,edge_ index = edge_ index,spa_pos = spa_pos)
File "/root/miniconda3/1ib/python3 .8/site - packages/ torch/ nn/ modules/module.pY". line 1102， in - call_ impl
return forward_ call(*input, **kwargs)
TypeError: forward() got an unexpected keyword argument ' spa_ pos
然后我试着把spa_pos这个参数去掉了，就可以正常运行了，我看好像是因为GATConv中没有这个参数，这个是自己加上去的，希望向您请教下关于这部分代码应该怎么写？谢谢您！