Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Version dependence of pytorch #1

Open
gary-young opened this issue Sep 14, 2022 · 1 comment
Open

Version dependence of pytorch #1

gary-young opened this issue Sep 14, 2022 · 1 comment

Comments

@gary-young
Copy link

Hi! This work is excellent so I want to reproduce the codes as soon as possible. However, when I try to reproduce the codes, I find there are some bugs such as the parameter mismatch when loading checkpoint:

checkpoint['model']['graph'] is expected to be a tensor or tensor-like but get a graph.Graph

After trying, I thought maybe it was related to the version of PyTorch. After rolling back the PyTorch version from 1.12.1 to 1.10.1 (and reinstalling the torch-scatter), the bug is solved. I believe if you are willing to test this issue and give some instructions, it can help more people who love this job like me.

@KiddoZhu
Copy link
Contributor

Hi! Thanks for pointing this out. This is generally a compatibility issue between TorchDrug and PyTorch, since TorchDrug overloads a few PyTorch modules. We will test TorchDrug on newer PyTorch versions and fix that.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants