Error running example branching-imitation.ipynb

[dxyzx0]

I installed all the packages required for this example. Here's my version of packages:

But when I run the code, I get the following error:

Why data return NoneType?
Thank you very much!

[dxyzx0]

Seems the problem lies in the incompatibility of torch_geometric=2.x.x and torch_geometric=1.x.x. I downgrade pyg to version 1.7.2 with pip install torch-geometric==1.7.2 and the problem solved.

PS: torch-sparse torch-scatter torch-cluster can be installed without *cuda.so in their corresponding folders, which will cause error when you use cuda.

Install them with option --no-index and the full command will be pip install torch-scatter --no-index -f https://pytorch-geometric.com/whl/torch-1.9.0+cu111.html