X Tutup
The Wayback Machine - https://web.archive.org/web/20200921184904/https://github.com/DeepGraphLearning/graphvite/issues/45
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

Load pre-trained models #45

Open
gustavopuma opened this issue Feb 21, 2020 · 1 comment
Open

Load pre-trained models #45

gustavopuma opened this issue Feb 21, 2020 · 1 comment

Comments

@gustavopuma
Copy link

@gustavopuma gustavopuma commented Feb 21, 2020

I’m using one virtual machine with 6 CPU / 1 GPU (11GB) with 56 GB RAM, I’m facing memory allocation error to load the pre-trained model Wikidata5m, using the python interface, can you explain why?

@KiddoZhu
Copy link
Collaborator

@KiddoZhu KiddoZhu commented Feb 21, 2020

It should be fine to load the pretrained model dump. I guess the reason is in the training stage.

The auto memory allocation mechanism doesn't work well in virtual matchines. If you specify auto for epsiode size, it would allocate w.r.t. the physical memory size. You may manually specify a value for episode size. Larger values train faster.

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

Successfully merging a pull request may close this issue.

None yet
2 participants
You can’t perform that action at this time.
X Tutup