search

weiyinwei / MMGCN

MMGCN: Multi-modal Graph Convolution Network forPersonalized Recommendation of Micro-video
280 stars 52 forks source link

关于GPU使用的疑惑 #53

Open chengyuehuang511 opened 2 years ago

chengyuehuang511 commented 2 years ago

作者您好，我看到您的代码里并没有用GPU，我将代码改成使用cuda会出现oom问题，但直接用CPU训练又非常慢。请问您是怎么解决这个问题的呢？