text_gcn Training 20ng dataset using GPU

Training 20ng dataset using GPU

Open ixnuhs opened this issue 5 years ago • 3 comments

Hi Dr. Yao,

I am curious if you tried to train 20ng dataset by GPU? I got the error below which seems like an exception thrown from tensforflow source code.

Cannot use GPU when output.shape[1] * nnz(a) > 2^31

thanks

Jun 20 '19 20:06 ixnuhs

Hi, @shun61

I tried this and it's an memory issue, the whole graph is loaded into GPU memory. smaller graphs like mr and R8 graph can work on GPU.

Jun 22 '19 04:06 yao8839836

Thank you.

Jun 22 '19 17:06 ixnuhs

Hello @yao8839836 , I got the same issue when I try to run the model with my own dataset. Could you please attach your system information?

Feb 21 '20 22:02 CharleoY