search

snap-stanford / relbench

RelBench: Relational Deep Learning Benchmark
https://relbench.stanford.edu
MIT License
220 stars 41 forks source link

Issue with running `train_model.ipynb` in Colab #274

Closed cpondoc closed 2 weeks ago

cpondoc commented 2 weeks ago

I'm working on a Relbench tutorial for the class CS 224W. I tried running the train_model.ipynb Colab but unfortunately got a common error across PyG:

'NeighborSampler' requires either 'pyg-lib' or 'torch-sparse'

After perusing through several threads on the issue, it appears as though it has something to do with installing certain dependencies (i.e., torch_scatter, torch_sparse) prior to installing torch_geometric. Using these tips, I was able to debug and get a version running locally (using a standalone Conda environment) as well on Colab (which I've made public here). Happy to make the PR but just thought I'd bring up the issue first!

cc @JosephGuman

rishabh-ranjan commented 2 weeks ago

Thanks for your efforts. A PR would be appreciated!