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

reaction prediction #2

Closed
WYejian opened this issue Mar 12, 2022 · 3 comments
Closed

reaction prediction #2

WYejian opened this issue Mar 12, 2022 · 3 comments

Comments

@WYejian
Copy link

WYejian commented Mar 12, 2022

The model throws the error "RuntimeError: CUDA error: device-side assertion fired" (graph2seq_series_rel.py
,line 125, in forward, memory_lengths=memory_lengths) when I run reaction prediction, not when I run retrosynthesis.

@DeepAIDD
Copy link

I also have this problem. Have you solved it

@zhengkaitu
Copy link
Collaborator

@WYejian @zpking Can you try to see whether the problem can be resolved by setting REL_BUCKETS to 11 in train_g2s.sh? CUDA device-side assertion typically points to the case when the lookup index is greater than what the embedding table contains.

@DeepAIDD
Copy link

@zhengkaitu Thank you very much. The problem has been solved

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

3 participants