When I train a more huge dataset by model RotatE, always show the question "CUDA out of memory" ?
Dionysusssss opened this issue · 1 comments
Dionysusssss commented
When I train a more huge dataset by model RotatE, always show the question "CUDA out of memory" ?
Edward-Sun commented
You may need to use a smaller hidden dimension size or use less negative samples during training.