-
Notifications
You must be signed in to change notification settings - Fork 627
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
LightGCN model,使用 GPU训练, GPU模式下加载预测报错, CPU模式下加载预测正常 #834
Comments
能否提供一下报错信息,以及yaml文件 |
|
使用GPU训练,然后GPU加载预测,就会爆出如上错误,改成CPU加载并预测没有错误。 |
你这个报错看上去是因为 |
但是问题是,切换到cpu模式就没有问题 |
@dxjjhm 你好,你提供的信息非常有限,我们无法进行bug复现,请你提供完整代码和必要的复现信息 |
I got the same error. I only add "repeatable: True" to the |
@zrg1993 Hello, I modify the |
I found that this error happened when one item only have one interation with one user in the dataset. So I think this case results into a error in |
@Sherry-XLL Many thanks. It's working well now. |
Describe the bug
A clear and concise description of what the bug is.
To Reproduce
Steps to reproduce the behavior:
Expected behavior
A clear and concise description of what you expected to happen.
Screenshots
If applicable, add screenshots to help explain your problem.
Colab Links
If applicable, add links to Colab or other Jupyter laboratory platforms that can reproduce the bug.
Desktop (please complete the following information):
The text was updated successfully, but these errors were encountered: