search

DeepGraphLearning / KnowledgeGraphEmbedding

MIT License
1.24k stars 264 forks source link

When I train a more huge dataset by model RotatE, always show the question "CUDA out of memory" ? #44

Closed Dionysusssss closed 3 years ago

Edward-Sun commented 3 years ago

You may need to use a smaller hidden dimension size or use less negative samples during training.