geometry-aware-transformer

Code repository for Geometry-aware Transformer (GeoT)

Dataset and data pipelines are based on Dimenet.

Provides model and training example with QM9 dataset.

Installation

This project manages dependency based on poetry.

poetry install

Training

  • specify training hyperparameter by hanling train_seml.py
poetry run train_seml.py