Running error with sampler.py in citation2

snap-stanford / ogb

Benchmark datasets, data loaders, and evaluators for graph machine learning

https://ogb.stanford.edu

MIT License

1.93k stars 397 forks source link

Running error with sampler.py in citation2 #215

Closed StephanieDU closed 3 years ago

StephanieDU commented 3 years ago

Hi, I am trying to run the code of sampler.py for the citation2 dataset. If GPU was enabled, it would end up with a runtime error "RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)". If CPU was used, it would raise an index out of bounds error " File "sampler.py", line 144, in train h = model(x[n_id], adjs) IndexError: index 10139841 is out of bounds for dimension 0 with size 2927963 " Any help would be appreciated!

rusty1s commented 3 years ago

Thanks for reporting. I fixed the example script.