Bug fix in GatedGRNN device

alelab-upenn / graph-neural-networks

Library to implement graph neural networks in PyTorch

GNU General Public License v3.0

243 stars 61 forks source link

Open camilomarino opened 3 years ago

camilomarino commented 3 years ago

Sends qhat and qcheck to the proper device.

Reference Issues/PRs Fixes #6