Skip to content

Latest commit

 

History

History

efficiency-comparison

Efficiency Comparison Experiment

Installation

conda create --name gaunt-ef python=3.10
conda activate gaunt-ef
pip install torch==1.13.1+cu117 -f https://download.pytorch.org/whl/torch_stable.html
pip install e3nn
pip install ipykernel

Usage

  • Efficiency comparison experiment for Equivariant Feature Interaction: test_equi_feat.ipynb
  • Efficiency comparison experiment for Equivariant Convolution: test_equi_conv.ipynb
  • Efficiency comparison experiment for Equivariant Many-Body Interaction: test_equi_many-body.ipynb