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

cuda out of memory #14

Open
Wjt0403 opened this issue Mar 23, 2023 · 1 comment
Open

cuda out of memory #14

Wjt0403 opened this issue Mar 23, 2023 · 1 comment

Comments

@Wjt0403
Copy link

Wjt0403 commented Mar 23, 2023

I tried to train a model on my own dataset, but I came with the problem "cuda out of memory", is there any good way to solve that problem?

@Lee-zix
Copy link
Owner

Lee-zix commented Mar 27, 2023

The memory cost is influenced by many factors. The most important factor is the number of historical facts. If your dataset has so many facts at one timestamp and you use a long history. The "cuda out of memory" problem may appear. The simplest way to this problem is to decrease the length of history snapshots you modeling. And maybe adding a negative sampling strategy to RE-GCN is also a common solution.

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