gmm

Install dependencies:

1. It is recommended to install the dependencies in a conda environment.
2. Install PyTorch, PyTorch Geometric, pytorch-lightning.

Run the example:

# Make gmm importable
export PYTHONPATH=$PWD

# Enter the `examples` directory
cd examples

# Run the training script
python train.py

# Test model performance
python test.py