snap-stanford/neural-subgraph-learning-GNN

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

prey176 opened this issue · 2 comments

prey176 commented

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

Thanks for bringing up the issue. This seems to be a versioning issue with python/pytorch/pytorch geometric -- pyg-team/pytorch_geometric#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

Thanks, I will look into it.