search

rusty1s / embedded_gcnn

Embedded Graph Convolutional Neural Networks (EGCNN) in TensorFlow
MIT License
78 stars 13 forks source link

pytorch #3

Open tonyandsunny opened 4 years ago

tonyandsunny commented 4 years ago

Thank you very much for your work, I am not very familiar with the tensorflow version, do you have this code for pytorch?

rusty1s commented 4 years ago

Except for the pre-processing part (which is pure numpy), this model can be easily implemented with PyTorch Geometric (e.g. see mnist_graclus.py). The pre-processing segmentation/adjacency/feature computation can be found here. After that, you just need to convert it to the PyG format :)