BiGCN icon indicating copy to clipboard operation
BiGCN copied to clipboard

Published 20 hours ago • verified publisher icon TianBian95

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

Open SamJJJj opened this issue 4 years ago • 2 comments

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

SamJJJj avatar Jan 28 '21 05:01 SamJJJj

me too

JolsonTsci avatar Nov 17 '21 08:11 JolsonTsci

me too.Is there a solution?

youran521 avatar Dec 02 '21 00:12 youran521