Unified-LTH-GNN
Unified-LTH-GNN copied to clipboard
[ICML 2021] "A Unified Lottery Tickets Hypothesis for Graph Neural Networks", Tianlong Chen*, Yongduo Sui*, Xuxi Chen, Aston Zhang, Zhangyang Wang
Hi! I am trying to run the pruning for OGB datasets. I keep getting this error. Any idea how to resolve this? It runs for 1 epoch for 10 subgraphs...
In the paper, you mentioned rewinding GNN's weights (in a while loop). For the final stage, will you use the initial weights or the last round of trained weights in...
Hi @Tianlong-Chen @yongduosui Thank you for this nice work! Could you please explain how you compute MACs for each model? Is there any related toolbox? Thank you! Best, Yongcheng
Hi, it seems that you have used DGL for comparatively smaller graphs (Cora, Citeseer, and PubMed) and PyTorch Geometric for graphs. Is there any specific reason for that? Please advise...