search

enoche / MMRec

A Toolbox for MultiModal Recommendation. Integrating 10+ Models...
GNU General Public License v3.0
367 stars 46 forks source link

Some problem about layergcn #21

Closed wenyu1009 closed 1 year ago

wenyu1009 commented 1 year ago

When I run layergcn， https://github.com/enoche/MMRec/blob/58cbd75f0e7a333e91fd12c51538a4d13ed678c2/src/models/layergcn.py#L56-L61

When we train the model using GPU, and enter line 58. An error will be reported. The parameters are not on the same device, one GPU and one CPU. When we remove .to(self.device), the problem is solved.

enoche commented 1 year ago

Thanks for your feedback. Yes. removing .to(self.device) in random setting is correct, will update soon~