cynricfu / MAGNN

Metapath Aggregated Graph Neural Network for Heterogeneous Graph Embedding
388 stars 68 forks source link

OOM kill #41

Closed knightzhong closed 7 months ago

knightzhong commented 1 year ago

Thanks for sharing the code! When I use other nodes in the DBLP dataset for preprocessing to search for meta path instances, such as some operations like the preprocess_DBLP.ipynb file. But it has always required a large amount of memory. Do the author know the reason for this?

cynricfu commented 1 year ago

Yes. preprocess_DBLP.ipynb does require a large amount of memory. You may just download and use the preprocessed dataset. Or you can revise the get_metapath_neighbor_pairs function to do some sub-sampling and reduce the dataset