You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Some variables can be deleted to optimize GPU memory use.
For example, in the implementation of TransR, since the parameters pretrained on TransE are set to TransR, the tensors occupied by TransE can be released by deleting the TransE model and cleaning CUDA cache. According to my test, this can save significant GPU memory space when it comes to large datasets.
The text was updated successfully, but these errors were encountered:
Some variables can be deleted to optimize GPU memory use. For example, in the implementation of TransR, since the parameters pretrained on TransE are set to TransR, the tensors occupied by TransE can be released by deleting the TransE model and cleaning CUDA cache. According to my test, this can save significant GPU memory space when it comes to large datasets.
Hey, I think we can communicate with openKE, the author doesn't reply for a long time, my qq is 1575797665
Some variables can be deleted to optimize GPU memory use.
For example, in the implementation of TransR, since the parameters pretrained on TransE are set to TransR, the tensors occupied by TransE can be released by deleting the TransE model and cleaning CUDA cache. According to my test, this can save significant GPU memory space when it comes to large datasets.
The text was updated successfully, but these errors were encountered: