Comments (3)
I believe this is a bug for other models as well. I'm running a TextRepresentation + DistMult interaction model and despite having 80G of VRAM PyKEEN still tries to allocate 14.90G more than I have. Conincidentally that's OOM by exactly the same margin as in your example.
from pykeen.
Hi @bolak92 ,
could you try whether #1261 has solved your issue? It's not yet in a release but you can use it by installing from source
pip install git+https://github.com/pykeen/pykeen.git
from pykeen.
I can confirm that I get this same issue on the latest version, when using apple silicon/"mps" device.
i.e consistently crashes on evaluation due to OOM when using "mps" (Macbook Pro M3).
from pykeen.
Related Issues (20)
- Importing model_resolver HOT 2
- Getting Embeddings of the Entity and Relations HOT 13
- RGCN Hyper parameter optimization error HOT 1
- MatKG HOT 1
- HPO_Pipeline fails on AutoSF models HOT 1
- Unable to reproduce TransE experiment
- EarlyStopper: show progress bar
- Cosine Annealing with Warm Restart LR Scheduler recieving an unexpected kwarg `T_i` HOT 1
- OOM Crash on MPS/Apple silicon HOT 2
- Reason for omitting validation inference triples from filtering when doing test evaluation in inductive lp example HOT 2
- tqdm progressbar is still shown although setting `use_tqdm=False`
- create_inverse_triples=True fails for the ILPC datasets
- pip installation broken (1.8.1) HOT 2
- `LabelBasedInitializer` sample code may be missing an argument `triples_factory` HOT 2
- issue in "Full Inductive LP Example" HOT 4
- "Class Inheritance Diagram" of documentation about prediction in pyKeen does not show correctly HOT 2
- Splitting a triples factory could not find a coverage of all entities and relations HOT 5
- List of mode's best performance config for each dataset
- Hit@10 does not match with the evaluator result
- Deprecated get_tail_prediction_df function in Understanding the Evaluation tutorial HOT 2
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from pykeen.