search

beta-team / beta-recsys

Beta-RecSys: Build, Evaluate and Tune Automated Recommender Systems
https://beta-recsys.readthedocs.io/en/latest/
MIT License
162 stars 33 forks source link

Update sgl.py #420

Closed TedSIWEILIU closed 2 years ago

TedSIWEILIU commented 2 years ago

Slove the issue when running SGL on GPU. Reason: self.adj_mat is not placed on the same device as ego_embeddings.