search

benedekrozemberczki / SimGNN

A PyTorch implementation of "SimGNN: A Neural Network Approach to Fast Graph Similarity Computation" (WSDM 2019).
GNU General Public License v3.0
759 stars 147 forks source link

GPU support #22

Closed shi27feng closed 4 years ago

shi27feng commented 4 years ago

Hi, I'm wondering whether this code supports running on GPUs? Thanks

benedekrozemberczki commented 4 years ago

You can do a transfer on top of the model and dataset.