GatedGRNN device error - Githubissues

alelab-upenn / graph-neural-networks

Library to implement graph neural networks in PyTorch

GNU General Public License v3.0

250 stars 64 forks source link

GatedGRNN device error #6

Open camilomarino opened 3 years ago

camilomarino commented 3 years ago

GatedGRNN does not send all the parameters to the correct device.

https://github.com/alelab-upenn/graph-neural-networks/blob/26c4d023f10a213a9ae13a6893a489b316c056b4/alegnn/utils/graphML.py#L1292