search

snap-stanford / neural-subgraph-learning-GNN

329 stars 64 forks source link

_pickle.PicklingError: Can't pickle typing.Union[torch.Tensor, NoneType] #2

Open prey176 opened 3 years ago

prey176 commented 3 years ago

We are working on Google Colab and were facing this issue, we were trying to train the model as stated in the READ ME with the following command ("python3 -m subgraph_matching.train --node_anchored"). The full error is stated as "_pickle.PicklingError: Can't pickle typing.Union[torch.Tensor, NoneType]: it's not the same object as typing.Union". We have also tried on the Colab GPU, but was getting the same error.

qema commented 3 years ago

Thanks for bringing up the issue. This seems to be a versioning issue with python/pytorch/pytorch geometric -- https://github.com/rusty1s/pytorch_geometric/issues/1541 may be related?

The code has been tested with Python 3.7.4, PyTorch 1.4.0 and Pytorch Geometric 1.4.3. If you are using the latest version of the libraries, upgrading Python may help?

prey176 commented 3 years ago

Thanks, I will look into it.