Back to Pytorch Geometric

Examples for PyTorch Lightning

examples/pytorch_lightning/README.md

2.7.0848 B
Original Source

Examples for PyTorch Lightning

This directory provides examples showcasing the integration of PyG with PyTorch Lightning.

ExampleDescription
graph_sage.pyCombines PyG and PyTorch Lightning for node classification via the GraphSAGE model
gin.pyCombines PyG and PyTorch Lightning for graph classification via the GIN model
relational_gnn.pyCombines PyG and PyTorch Lightning for heterogeneous node classification