Closed Aaricis closed 10 months ago
Hi!
It's likely that you did not convert COO edge_index
into SparseTensor
format. See here. The following should work.
import torch_geometric.transforms as T
from ogb.nodeproppred import PygNodePropPredDataset
dataset = PygNodePropPredDataset(name='ogbn-arxiv', transform=T.ToSparseTensor())
I loaded
ogbn-arxiv
, and then trained this dataset withGCN
model. I encontered this error, but I donnot know why. my environment: