PyG中自定义Data的注意事项(__cat_dim__)

记录一个超超超超超令人无语的事件

下载PyG跑源码,然后使用DataLoader加载自定义的数据,结果报错:

__cat_dim__() takes 3 positional arguments but 4 were given

找不到原因,看了一下博客,GNN学习笔记(八):超大规模数据集类的创建、图预测任务实践_唠叨小主的博客-CSDN博客_二部图数据集

将其中第5节的代码拿下来跑,也是报的一样的错误

实在没有办法,在本地电脑上配好了环境,打算用pycharm一步一步debug

结果代码一粘贴进来就出现了警告:

重写的方法没有对应上!!!!!!!

最后找到BaseData类的原始方法:

 

修改重写方法的参数表:

 

 然后就可以了

应该是版本不一致导致的,太搞了。。。 


版权声明:本文为tagagi原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。