2 - Priority P2benchmarkfeaturehelp wantednn
Repository metrics
- Stars
- (19,985 個のスター)
- PR merge metrics
- (平均マージ 35d 1h) (30d で 14 merged PRs)
説明
🐛 Bug
I have benchmarked EEC vs https://github.com/mys007/ecc implementation. The code is 3-5 times slower in pytorch_geometric.
I think it is due to the entire pseudo tensor which is given to every message function. Maybe, only weight_j should be given there.
To Reproduce
Steps to reproduce the behavior:
Expected behavior
Environment
- OS:
- Python version:
- PyTorch version:
- CUDA/cuDNN version:
- GCC version:
- Any other relevant information: