search

torchmd / torchmd-net

Training neural network potentials
MIT License
335 stars 75 forks source link

Remove dependency on torch_cluster #228

Closed RaulPPelaez closed 1 year ago

RaulPPelaez commented 1 year ago

This PR replaces every usage of the Distance module with the OptimizeDistance one. I moved the old Distance module to the benchmarks/neighbors.py.