search

ycq091044 / SafeDrug

IJCAI2021: Code for SafeDrug, MIMIC data processing, Medical code mapping
67 stars 16 forks source link

MPNN module is not be trained in the training stage. #3

Closed xlbryantx closed 2 years ago

ycq091044 commented 2 years ago

I directly assign the output to "device", which should address the issue. Thanks for pointing out!