flairNLP / flair

A very simple framework for state-of-the-art Natural Language Processing (NLP)
https://flairnlp.github.io/flair/
Other
13.85k stars 2.09k forks source link

[Question]: How much GPU Memory For Zelda Entity Linking? #3433

Open zrjohnnyl opened 6 months ago

zrjohnnyl commented 6 months ago

Question

Hi, I'm trying to run train an entity linking model using the zelda dataset but I'm getting a CUDA out of memory error. I'm currently using V100 GPUs. I'm wondering what is the minimum GPU memory you need to train this model on this Zelda dataset?