Error when running GNN-seg

Xi-yuanWang / GLASS

GLASS: GNN with Labeling Tricks for Subgraph Representation Learning

30 stars 6 forks source link

Error when running GNN-seg #3

Closed shhs29 closed 2 years ago

shhs29 commented 2 years ago

Hi,

I am getting this error when I am running GNNSeg: TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, dicts or lists; found <class 'torch_geometric.data.data.Data'>

This error occurs during the model training. Kindly let me know if you have any idea what might have caused this and how I can fix it.

Thanks and Regards, Shweta Ann Jacob

shhs29 commented 2 years ago

Hi,

I am closing this issue as this was an error on my machine due to incorrect import of DataLoader.

Thanks and Regards, Shweta Ann Jacob