Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/tchaton/lightning-geometric
Integrate pytorch
https://github.com/tchaton/lightning-geometric
Last synced: 3 months ago
JSON representation
Integrate pytorch
- Host: GitHub
- URL: https://github.com/tchaton/lightning-geometric
- Owner: tchaton
- Created: 2020-09-11T17:34:52.000Z (about 4 years ago)
- Default Branch: master
- Last Pushed: 2021-04-04T09:23:50.000Z (over 3 years ago)
- Last Synced: 2024-07-04T02:11:46.213Z (4 months ago)
- Language: Python
- Size: 1.77 MB
- Stars: 47
- Watchers: 2
- Forks: 2
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# TorchScripted Pytorch Geometric Examples with Pytorch Lightning and Hydra
[](https://codecov.io/gh/tchaton/lightning-geometric) [](https://github.com/tchaton/lightning-geometric/actions)
### Setup on MacOs. Please, adapt to others OS :)
```python
brew install cmake
pyenv install 3.7.8
pyenv local 3.7.8
python -m venv
source .venv/bin/activate
poetry install
```
### PRINCIPAL CMD
```python
python train.py model={{MODEL}} dataset={{DATASET}} loggers={{LOGGERS}} log={{LOG}} notes={{NOTES}} name={{NAME}} jit={{JIT}}
```
- `LOGGERS` str: Configuration file to log to Wandb, currently using mine as `thomas-chaton`
- `LOG` bool: Wheter to log training to wandb
- `NOTES` str: A note associated to the training
- `NAME` str: Training name appearing on Wandb.
- `LOG` bool: Wheter to make model jittable.
### Working Inference
Have a look at `test/test_inference.py`
### SUPPORTED COMBINAISONS
| `{{DATASET}}` | `{{MODEL}}` | DATASET DESCRIPTION | MODEL DESCRIPTION | WORKING | |
| ------------- | ----------- | ----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- | ------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------ | ---------------------------- | --- |
| zinc | pna | The ZINC dataset from the "Grammar Variational Autoencoder" | The Principal Neighbourhood Aggregation graph convolution operator from the "Principal Neighbourhood Aggregation for Graph Nets" | True |
| faust | spline | The FAUST humans dataset from the "FAUST: Dataset and Evaluation for 3D Mesh Registration" | The spline-based convolutional operator from the "SplineCNN: Fast Geometric Deep Learning with Continuous B-Spline Kernels" | In progress |
| ppi | gat | The protein-protein interaction networks from the "Predicting Multicellular Function through Multi-layer Tissue Networks" | The graph attentional operator from the "Graph Attention Networks" True | True |
| cora | agnn | The citation network datasets "Cora", "CiteSeer" and "PubMed" from the "Revisiting Semi-Supervised Learning with Graph Embeddings" | "Attention-based Graph Neural Network for Semi-Supervised Learning" | True |
| cora | sage | "" | The GraphSAGE operator from the "Inductive Representation Learning on Large Graphs" | True |
| cora | sgc | "" | The simple graph convolutional operator from the "Simplifying Graph Convolutional Networks" | True |
| cora | tag | "" | The topology adaptive graph convolutional networks operator from the "Topology Adaptive Graph Convolutional Networks" | True |
| cora | dna | "" | The dynamic neighborhood aggregation operator from the "Just Jump: Towards Dynamic Neighborhood Aggregation in Graph Neural Networks" | True |
| reddit | sage | The Reddit dataset from the "Inductive Representation Learning on Large Graphs" | "" | True |
| reddit | agnn | "" | "" | True |
| icews18 | renet | The Integrated Crisis Early Warning System (ICEWS) dataset used in the, _e.g._, "Recurrent Event Network for Reasoning over Temporal Knowledge Graphs" | The Recurrent Event Network model from the "Recurrent Event Network for Reasoning over Temporal Knowledge Graphs" | Waiting for support for TGCN |
| cora | argva | "" | The Adversarially Regularized Variational Graph Auto-Encoder model from the "Adversarially Regularized Graph Autoencoder for Graph Embedding" ` | True |
| cora | arma | "" | The ARMA graph convolutional operator from the "Graph Neural Networks with Convolutional ARMA Filters" .01343> | True |
| cora | gcn | "" | The GCN graph convolutional operator from the "Semi Supervised Classification with Graph Convolution Networks" .01343> | True |
| cora | gcn2 | "" | The graph convolutional operator with initial residual connections and identity mapping (GCNII) from the "Simple and Deep Graph Convolutional Networks" | True |
# DATASET SIZES
```
529M ./Flickr
74M ./FAUST
16M ./cora
3.5G ./Reddit
383M ./ZINC
1.8G ./MNISTSuperpixels
182M ./OgbnArxiv
192M ./PPI
156M ./ICEWS18
6.8G .
```