I have trouble in running your code(CUDA out of memory error)

TianBian95 / BiGCN

Source Codes: Rumor Detection on Social Media with Bi-Directional Graph Convolutional Networks--AAAI 2020

250 stars 73 forks source link

I have trouble in running your code(CUDA out of memory error) #7

Open SamJJJj opened 3 years ago

SamJJJj commented 3 years ago

I have tried to reduce the batch_size to 2, but I can only run a few of epochs, and I also tried to add th.cuda.empty_cache() in your code. I run this: python ./model/Weibo/BiGCN_Weibo.py 100

and my envirment is : GPU : RTX3080. pytorch 1.4.0 CUDA10.1

JolsonTsci commented 2 years ago

me too

youran521 commented 2 years ago

me too.Is there a solution?