mims-harvard / Raindrop

Graph Neural Networks for Irregular Time Series
MIT License
161 stars 33 forks source link

hard coded cuda path and undeclared pytorch dependency #2

Open tsonika opened 2 years ago

tsonika commented 2 years ago

Hi There,

I am interested in using your model but ran into a couple of issues:

  1. lines 8-9 in models_rd.py have a hardcoded path to a windows package.
  2. there is an undeclared dependency pytorch_geometric

I was wondering if you can make some suggestions on the above issues. many thanks Sonika