BUPT-GAMMA / OpenHGNN

This is an open-source toolkit for Heterogeneous Graph Neural Network(OpenHGNN) based on DGL.
Apache License 2.0
848 stars 142 forks source link

[Fix]: GTN, fastGTN, MHNF model 'RuntimeError' when using gpu #154

Closed bupt229 closed 1 year ago

bupt229 commented 1 year ago

Description

Fix the 'RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)' problem for [GTN] [fastGTN] [MHNF] models.

Checklist

Please feel free to remove inapplicable items for your PR.

Changes