Alby M.
Thanks @frobnitzem! Will be very happy to merge and have this feature. One small thing: can you run the code through `black`: https://nequip.readthedocs.io/en/develop/dev/contributing.html#style-enforcement? And one bigger thing: > The TensorBoard...
Sounds great, thanks!
Hi @wkylee14 , Thanks for your interest in our code! `stress` should use a normal, and not `PerAtom` loss. Stress training issues are often linked to incorrect labels, either due...