-
Notifications
You must be signed in to change notification settings - Fork 3.4k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[Example] Add Pytorch Geometric Example (#4568)
* add example for Pytorch Geometric * remove hydra * add docstring * remove description * rename folder * update script to not break test * remove .lock * add Pytorch Geometric to doc * add docstring at the begining * add comments * Update pl_examples/pytorch_ecosystem/pytorch_geometric/README.md Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com> * Update pl_examples/pytorch_ecosystem/pytorch_geometric/README.md Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com> * Update pl_examples/pytorch_ecosystem/pytorch_geometric/cora_dna.py Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com> * add toml Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com> Co-authored-by: Jirka Borovec <jirka@pytorchlightning.ai>
- Loading branch information
Showing
8 changed files
with
476 additions
and
2 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,4 @@ | ||
Pytorch Ecosystem Examples | ||
========================== | ||
|
||
- `Pytorch Geometric: Deep learning on Graphs and other irregular structures <https://github.com/PyTorchLightning/pytorch-lightning/tree/master/pl_examples/pytorch_ecosystem/pytorch_geometric/README.md>`_. |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,38 @@ | ||
# [Pytorch Geometric](https://github.com/rusty1s/pytorch_geometric) examples with Lighting | ||
|
||
### Introduction | ||
|
||
PyTorch Geometric (PyG) is a geometric deep learning extension library for PyTorch. It relies on lower level libraries such as | ||
|
||
* PyTorch Cluster: A package consists of a small extension library of highly optimized graph cluster algorithms in Pytorch | ||
* PyTorch Sparse: A package consists of a small extension library of optimized sparse matrix operations with autograd support in Pytorch | ||
* PyTorch Scatter: A package consists of a small extension library of highly optimized sparse update (scatter and segment) operations for the use in PyTorch | ||
|
||
## Setup | ||
|
||
``` | ||
pyenv install 3.7.8 | ||
pyenv local 3.7.8 | ||
python -m venv | ||
source .venv/bin/activate | ||
poetry install | ||
``` | ||
|
||
Run example | ||
|
||
``` | ||
python cora_dna.py | ||
``` | ||
|
||
## Current example lists | ||
|
||
| `DATASET` | `MODEL` | `TASK` | DATASET DESCRIPTION | MODEL DESCRIPTION | | | ||
| :---: | :---: | :---: | :---: | :---: | :---: | | ||
| Cora | DNA | Node Classification | The citation network datasets "Cora", "CiteSeer" and "PubMed" from the "Revisiting Semi-Supervised Learning with Graph Embeddings" <https://arxiv.org/abs/1603.08861> | The dynamic neighborhood aggregation operator from the "Just Jump: Towards Dynamic Neighborhood Aggregation in Graph Neural Networks" | ||
|
||
|
||
## DATASET SIZES | ||
|
||
``` | ||
16M ./cora | ||
``` |
Empty file.
Oops, something went wrong.