Requirement for the environment of models

[zouzh14]

Hello,

I want to use your model to optimize some ligands and I install a conda env with torch and torch_geometric. When I run the example of 1nja, it reports the error:

AttributeError: 'TransformerConv' object has no attribute '_propagate_forward_pre_hooks'

It seems related with the different version of pyg between the saved model and the current env. What are your torch and torch_geometric versions? My env is torch 1.10.0, torch.cuda 11.3, torch_geometric 2.0.2.