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

GPU Memory Increases Significantly #6

Open
zyouyang opened this issue Apr 25, 2023 · 3 comments
Open

GPU Memory Increases Significantly #6

zyouyang opened this issue Apr 25, 2023 · 3 comments

Comments

@zyouyang
Copy link

Hi,

Thanks for sharing the codes. Your work is quite interesting. However, as I was reproducing the experiments, I found that every time after performing the iterative training, the GPU memory increases significantly and will not drop back to what it was before the iterative training. This might cause GPU out of memory issues. Would you investigate this problem and see if it is reproducible on your side?

b/w: could you also share the hyperparameters for other datasets such as FB15K237-20? Thanks!

Best,
Zhongyu

@bys0318
Copy link
Owner

bys0318 commented Apr 26, 2023

I will try to navigate the problem about GPU usage when activating iterative training. The hyperparameters on all 4 datasets are shown in the command line in README.md, you will reproduce our reported results by running these commands:)

@bys0318
Copy link
Owner

bys0318 commented Apr 26, 2023

BTW, you can also find the best hyperparameters in Table 8 of our paper (appendix).

@zyouyang
Copy link
Author

Sorry for missing them. Thanks for letting me know!

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