about running time
sana-mine opened this issue · 4 comments
Hello, I am interested in your paper. I would like to ask you about the approximate time for this code to run in each task. How much GPU memory is consumed?
It depends on the data size. For dataset such as WN18RR or FB15K237, the whole procedure usually take one or two days in one 3090 gpu (24 GB).
It depends on the data size. For dataset such as WN18RR or FB15K237, the whole procedure usually take one or two days in one 3090 gpu (24 GB).
Thanks. But I used your default parameters to test on 'umls' dataset, and found that there is a certain gap between the running results and the paper data. Is there any modification in other parts?
Hi, it is caused by the random sampling strategy or pre-training procedure, especially for the small dataset such as UMLS. You can try to pre-train the task with more epochs or use our pre-training checkpoints (which will be attached in this repository).
Sorry for the inconvenience.