search

diningphil / gnn-comparison

Official Repository of "A Fair Comparison of Graph Neural Networks for Graph Classification", ICLR 2020
GNU General Public License v3.0
385 stars 54 forks source link

Can't find device = cuda in config files #43

Closed YoavLotem closed 1 year ago

YoavLotem commented 1 year ago

Hi team :) Thanks for this great work!

I was wondering why doesn't the configuration files contain cuda as a possible device option even though the paper mentions it as a possible device?

For example here are two images from GraphSAGE config file and from the paper: image image

Thank you in advance! @Zmeos @diningphil @marcopodda @dbacciu

diningphil commented 1 year ago

Hi,

If you want to use cuda, just change the configuration file from cpu to cuda. We did not specify it in the public file because we do not know if your pc has cuda installed.

Best regards, Federico