BorgwardtLab / TOGL

Topological Graph Neural Networks (ICLR 2022)
https://openreview.net/pdf?id=oxxUMeFwEHd
BSD 3-Clause "New" or "Revised" License
105 stars 20 forks source link

Segmentation fault occurs when I try to insert the topologyLayer to my own GNN module #18

Open ljj-cyber opened 3 months ago

ljj-cyber commented 3 months ago

I want to import your topology layer to my own neural network called GDN. But when I try to use the operation of graph batch, the segmentation fault occurs. Please let me know the input type of the TopologyLayer. So that I can adjust my code. Thanks a lot. There is some detail of my code below: image image This code works well. But if I use 'Batch' in pyg to encapsulate the graph data, it fails. image The segmentation fault occurs in the following position(#116). image