AntNLP / gnn-dep-parsing

Apache License 2.0
32 stars 7 forks source link

Question on the GPU Requirements for GNN Training #5

Closed matyascorvinus closed 3 years ago

matyascorvinus commented 3 years ago

Dear the Development Team, I want to ask what is the hardware you used to train the GNN with the PTB data? I use GTX 1060 6GB RAM and GTX 1080 TI 11GB RAM and the training proceeded until 50k iteration and the process killed it. I want to understand whether I need a better GPU or the environment has problems as well!

Thank you so much for your time and consideration! Dang Quang