The memory payload is very high?

zjukg / KGTransformer

[Paper][WWW2023] Structure Pre-training and Prompt Tuning for Knowledge Graph Transfer

https://arxiv.org/pdf/2303.03922.pdf

46 stars 5 forks source link

The memory payload is very high? #2

Closed WEIYanbin1999 closed 1 year ago

WEIYanbin1999 commented 1 year ago

When I run the pretrain code, I found the memory will be toughly full in a machine with 100+G memory. Why its CPU occupancy and memory requirements are so high? Is it normal?

Even I running in another machine with 30G memory, it is killed by OS because of its memory payload

WEIYanbin1999 commented 1 year ago

Could you please provide a link for downloading the pre-trained model parameters? I think it is much helpful for us to follow your work

YangL256 commented 1 year ago

I had the same problem. I ran it on the cloud server RTX 3090 for 11 minutes before it was killed

WEIYanbin1999 commented 1 year ago

The memory occupancy comes from the huge 2-hop triple data dict.