Can this code be run with multiple GPUs?

mims-harvard / SubGNN

Subgraph Neural Networks (NeurIPS 2020)

https://zitniklab.hms.harvard.edu/projects/SubGNN

MIT License

185 stars 33 forks source link

Can this code be run with multiple GPUs? #13

Open leaf-ygq opened 3 years ago

leaf-ygq commented 3 years ago

Hi @EmilyAlsentzer , I want to know if the code can be run with multiple GPUs. I change the parameter of trainer in train_config.py. But there was something wrong after I finished it. I'm sure that there is no bug when I use only one GPU. I really hope that you could help me deal with it! Please feel free to concat me.