Code repository for Geometry-aware Transformer (GeoT)
Dataset and data pipelines are based on Dimenet.
Provides model and training example with QM9 dataset.
This project manages dependency based on poetry.
poetry install
- specify training hyperparameter by hanling train_seml.py
poetry run train_seml.py