torchmd-net
torchmd-net copied to clipboard
MACEOFF dataset needs .get_atomref
I believe the MACEOFF dataset contains total energies, with the provided TensorNet-MACE.yaml it will not train properly. The losses diverge. It needs to be run with remove_ref_energy: true and a get_atomref() method implemented in datasets/maceoff.py