This directory contains code for downloading wandb metrics and generating plots for the paper 'Training Dynamics of Learning 3D-Rotational Equivariance' (TMLR, 2025).
Wandb metrics of equivariance metrics are generated by running training scripts in the escaip, proteina, and voxmol directories.
Cite as:
@article{
shen2025training,
title={Training Dynamics of Learning 3D-Rotational Equivariance},
author={Max W Shen and Ewa Nowara and Michael Maser and Kyunghyun Cho},
journal={Transactions on Machine Learning Research},
issn={2835-8856},
year={2025},
url={https://openreview.net/forum?id=DLOIAW18W3},
note={J2C Certification}
}