Commit 67dd31c4 authored by rusty1s's avatar rusty1s
Browse files

fix readme

parent c427300e
......@@ -17,6 +17,7 @@ It provides an easy-to-use interface to convert common and custom GNNs from PyG
```python
from torch_geometric.nn import GCNConv
from torch_geometric_autoscale import ScalableGNN
from torch_geometric_autoscale import metis, permute, SubgraphLoader
class GNN(ScalableGNN):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment